1

我有一个将一些 sql 放入数据框中的函数。为此,我正在使用 pandas,因此我的代码如下所示:

def get_all_sql_data(connection):

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

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

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

4

0 回答 0