Модульное тестирование сеанса SQLAlchemy

Question

Я пытаюсь создать тестовый скрипт для программы на Python, которая использует сеанс SqlAlchemy для подключения к базе данных. и получить данные из базы данных, используя session.query. Я не могу смоделировать файл session.query в скрипте модульного теста

Я попытался ввести значение сеанса с помощью UnifiedAlchemyMagicMock. Но я не смог добиться успеха в этом. Запросить помощь, если кто-то сделал насмешку

class TableClass(base):
    __tablename__ = 'table_name'
    __table_args__ = {'schema': 'dbschema', 'autoload': True, 'autoload_with': dbengine}

    session = sessionmaker(bind=dbengine)()
    def get_platforms_by_code(self,cd):

        result = self.session.query(TableClass).filter(TableClass.code==cd).first()


        return result