Модульное тестирование в Python. Можем ли мы имитировать результат SQL-запроса в Python? ⇐ Python

[Anonymous]

Резюме: у меня есть метод, который я хотел бы протестировать. Этот метод включает подключение базы данных, извлечение результата (из запроса sql) и сохранение его в фрейме данных.
Я хотел бы знать, есть ли способ издеваться над результат, полученный из запроса SQL?
Я попробовал библиотеку unittest, создал кадр данных, подобный полученному из запроса SQl. Но я не могу понять, как передать эту переменную в метод во время тестирования.
Поделитесь своими мыслями.
Мой метод выглядит примерно так:

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

    try:
db_connect = sqlite3.connect("address_zip.db")
db_cur = db_connect.cursor()
db_cur.execute(""" Select STD_ADDRESS,STD_CITY,STD_STATE from address""")
sql_result=pd.read_sql_query(""" Select STD_ADDRESS,STD_CITY,STD_STATE from address""",db_connect)
df = pd.DataFrame(sql_result, columns=['STD_ADDRESS', 'STD_CITY', 'STD_STATE'])
print(f"The data from database is :\n {df}")
db_cur.close()
db_connect.close()
except Exception as e:
print(f"DB exception: {e}")

return df

В исходном коде над переменной «df» выполняется несколько действий, моя цель — имитировать этот объект df.
Но я не могу понять, как?
Извините, это может быть очень простой вопрос, я новичок в Python.

Подробнее здесь: https://stackoverflow.com/questions/627 ... -in-python