在Python中替换测试中的变量(使用mock?或​​其他方式)

时间:2016-04-06 13:53:32

标签: python unit-testing testing mocking stubs

我有一个将一些sql放入数据帧的函数。对于它我使用熊猫,使我的代码看起来像这样:

def get_all_sql_data(connection):

    code = 'select * from table1'
    data = pd.read_sql(code, connection)
    ...
    ...
    ...
    return value

但是,由于我不想从测试代码中的数据库中获取数据,因此我希望在测试中用csv文件中的数据替换这些数据。

我知道如何使用模拟这样做,如果数据'是函数的显式参数。有没有办法让我用我想要使用的数据替换函数中的这个变量?如果没有,这个测试问题是否有更好的解决方法?

0 个答案:

没有答案