Python – Модульные тесты – Sql Alchemy – Automap ⇐ Python

[Anonymous]

Я пытаюсь написать модульные тесты для функций, выполняющих запросы Sql Alchemy. Я использую Automap для автоматического создания моделей. Поскольку у меня есть доступ к представлениям, а не к таблицам, я явно указал только первичный ключ моей модели. Ниже приведен мой код.
sql_alchemy_models.py

Код: Выделить всё

class Model1(Base):
__tablename__ = 'table_cust_data'

CustID = Column(Integer, primary_key=True)

main.py

Код: Выделить всё

from some_path.sql_alchemy_models import Model1

def get_info(session, cust_id):
cust_id = session.execute(select(Model1.CustID).filter(Model1.CustID == cust_id)).scalar()
return cust_id

test_main.py

Код: Выделить всё

from mock_alchemy.mocking import UnifiedAlchemyMagicMock

@pytest.fixture
def session():
return UnifiedAlchemyMagicMock()

def test_info(session, mocker):
mocker.patch('some_path.main.Model1')
session.execute.return_value.scalar.return_value = 1
result = get_info(session, 1)
assert result == 1

Когда я запускаю приведенный выше тест, я получаю эту ошибку: sqlalchemy.exc.ArgumentError: ожидаемое выражение столбца, предложение FROM или другой элемент предложения столбцов получили .
Может кто-нибудь поделиться, как устранить эту ошибку?

Подробнее здесь: https://stackoverflow.com/questions/792 ... my-automap