我有一个将一些sql放入数据帧的函数。对于它我使用熊猫,使我的代码看起来像这样:
def get_all_sql_data(connection):
code = 'select * from table1'
data = pd.read_sql(code, connection)
...
...
...
return value
但是,由于我不想从测试代码中的数据库中获取数据,因此我希望在测试中用csv文件中的数据替换这些数据。
我知道如何使用模拟这样做,如果数据'是函数的显式参数。有没有办法让我用我想要使用的数据替换函数中的这个变量?如果没有,这个测试问题是否有更好的解决方法?