我正在开发一组 python 脚本来预处理数据集，然后使用 scikit-learn 生成一系列机器学习模型。 我想开发一组单元测试来检查数据预处理功能，并希望能够使用一个小的测试熊猫数据框，我可以确定答案并在断言语句中使用它。

我似乎无法让它加载数据帧并将其传递给使用 self. 我的代码看起来像这样；

def setUp(self):
    TEST_INPUT_DIR = 'data/'
    test_file_name =  'testdata.csv'
    try:
        data = pd.read_csv(INPUT_DIR + test_file_name,
            sep = ',',
            header = 0)
    except IOError:
        print 'cannot open file'
    self.fixture = data

def tearDown(self):
    del self.fixture

def test1(self):    
    self.assertEqual(somefunction(self.fixture), somevalue)

if __name__ == '__main__':
    unittest.main()

谢谢您的帮助。

#1楼 票数：31 已采纳

Pandas 有一些用于测试的实用程序。

import unittest
import pandas as pd
from pandas.util.testing import assert_frame_equal # <-- for testing dataframes

class DFTests(unittest.TestCase):

    """ class for running unittests """

    def setUp(self):
        """ Your setUp """
        TEST_INPUT_DIR = 'data/'
        test_file_name =  'testdata.csv'
        try:
            data = pd.read_csv(INPUT_DIR + test_file_name,
                sep = ',',
                header = 0)
        except IOError:
            print 'cannot open file'
        self.fixture = data

    def test_dataFrame_constructedAsExpected(self):
        """ Test that the dataframe read in equals what you expect"""
        foo = pd.DataFrame()
        assert_frame_equal(self.fixture, foo)

#2楼 票数：22

如果您使用的是最新的熊猫，我认为以下方式更简洁：

import pandas as pd

pd.testing.assert_frame_equal(my_df, expected_df)
pd.testing.assert_series_equal(my_series, expected_series)
pd.testing.assert_index_equal(my_index, expected_index)

如果它们不“相等”，这些函数中的每一个都会引发AssertionError 。

有关更多信息和选项： https : //pandas.pydata.org/pandas-docs/stable/reference/general_utility_functions.html#testing-functions

#3楼 票数：0

你也可以用snapshottest做这样的事情。

https://stackoverflow.com/a/64070787/3384609

  ask by tjb305 translate from so

未解决问题？本站智能推荐：

1回复

使用 unittest 框架测试 Pandas 数据框

我正在尝试使用 python unittest框架进行处理 csv 文件的单元测试。 我想测试列名称匹配、列中的值匹配等情况。我知道有更方便的库，例如datatest和pytest ，但我只能在我的项目中使用unittest 。 猜猜我使用了错误的unittest.TestCase方法，并以错误的
1回复

如何对数据帧操作进行单元测试？

我有一个功能： 例如一个测试用例可以是：我创建一个数据框： 这是我尝试过的： 这给了我错误AssertionError: Series Expected type <class 'pandas.core.series.Series'>, found <class 'pa
1回复

这是在python中测试pandas数据框的正确方法吗？

我有一个函数，它返回一个包含行和列的数据框。 这是函数：app.py 这是我的单元测试用例：test_app.py 测试用例运行，但我是否以正确的方式测试它？
2回复

单元测试模拟熊猫 to_csv

我的模块.py test_mymodule.py 当我运行这个测试时，我得到： 我试图在我的方法中模拟to_csv 。 我的其他测试按预期运行，但是我不确定这个测试哪里出了问题。 我对 MagicMock 的使用是正确的，还是我忽略了其他东西？
1回复

无法识别 Python Pandas 单元测试

我有单元测试定义的问题。 我要测试数据帧，我不明白为什么返回以下结果。 结果： 脚本： import unittest import pandas as pd from pandas._testing import assert_frame_equal def df_minus(df_ma
2回复

将“ assertTrue”单元测试应用于熊猫DataFrame吗？

使用python进行单元测试，如何检查pandas DataFrame df列名与列表的列名一致？ 例如，我应该为此df为False：
3回复

Python单元测试模拟。 ValueError：DataFrame的真值不明确

我正在为我的python 2.7方法之一编写单元测试用例。 在我的测试方法中，有一个方法调用，该方法将字典包含字符串键和panadas数据帧作为该键的值。 我想为此方法编写一个交互测试，以检查它是否使用正确的字典在内部调用该方法 在单元测试中，我确实写了我的断言来测试这种交互
1回复

如何将assertSequenceEqual与pandas系列一起使用？

我想使用unittest模块检查两个pandas.Series对象是否具有相同的内容： 根据unittest文档，以上内容应该可以工作（基于docs ）。 但是，当我运行上述单元测试时，我得到了： 如何测试系列是否相同？ 请注意，将Series转换为list对象似乎可以解