我有一个将一些 sql 放入数据框中的函数。为此,我正在使用 pandas,因此我的代码如下所示:
def get_all_sql_data(connection):
code = 'select * from table1'
data = pd.read_sql(code, connection)
...
...
...
return value
但是,由于我不想在测试代码中从我们的数据库中获取我的数据,我希望在我的测试中用来自 csv 文件的数据替换这些数据。
如果“数据”是函数的显式参数,我知道如何使用模拟来做到这一点。有没有办法让我用我想使用的数据替换函数中的这个变量?如果没有,是否有更好的解决方法来解决这个测试问题?