在函数

时间:2018-06-06 17:32:19

标签: python pandas pytest

我正在尝试对一组函数运行单元测试。这些函数返回多个pandas数据帧。

如何使用pytest处理多个数据帧之间的相等性?

感谢您的帮助!

@pytest.fixture
def function_to_test():
    df1 = pd.DataFrame({'random': ['1'],'columns': ['One']})
    df2 = pd.DataFrame({'random': ['1'],'columns': ['One']})
    df3 = pd.DataFrame({'random': ['1'],'columns': ['One']})
    return df1, df2, df3


@pytest.fixture
def function_to_equal():
    df1e = pd.DataFrame({'random': ['1'],'columns': ['One']})
    df2e = pd.DataFrame({'random': ['1'],'columns': ['One']})
    df3e = pd.DataFrame({'random': ['1'],'columns': ['One']})
    return df1e, df2e, df3e


def unit_test_df-df3_function():
    #assert....?

0 个答案:

没有答案