是否有一个工厂帮助程序模块,可让我在以下位置重用/共享代码: 南数据迁移 TestCase.setUp() 我发现自己在这两个地方一次又一次地做同样的事情。 南数据迁移习惯于创建基本数据对象。 大多数情况下,它们在项目中(不在可重用的应用程序中)。 不幸的是, ...
提示:本站收集StackOverFlow近2千万问答,支持中英文搜索,鼠标放在语句上弹窗显示对应的参考中文或英文, 本站还提供 中文繁体 英文版本 中英对照 版本,有任何建议请联系yoyou2525@163.com。
我正在开发一组 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()
谢谢您的帮助。
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)
如果您使用的是最新的熊猫,我认为以下方式更简洁:
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
你也可以用snapshottest
做这样的事情。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.