单元测试 将变量作为参数传递

Question

我有一个数据结构，它在一个函数中创建并传递给其他函数。我正在尝试对所有这些功能进行单元测试。

我是否需要在每个函数中重新创建该数据结构（环境）？我尝试使用全局变量，但我无法保证哪个测试用例将在另一个测试用例之前运行。

我知道我不能不头疼地推翻unittest.TestCase __init__。

我还能如何实现这一目标？传递参数或以某种方式使其成为变量并避免竞争条件？

Answer 1

听起来您不想在每次测试之前重新定义数据结构。只要测试不修改数据，我认为在__init__中定义数据结构没有任何问题：

import unittest
class Test(unittest.TestCase):
    def __init__(self, methodName = 'runTest'):
        unittest.TestCase.__init__(self, methodName)
        self.data = range(5)
    def test_getitem(self):
        self.assertEqual(self.data[1],1)
    def test_reversed(self):
        self.assertEqual(list(reversed(self.data)),[4,3,2,1,0])
if __name__ == '__main__':
    import sys
    sys.argv.insert(1,'--verbose')
    unittest.main(argv = sys.argv)

收益 率

% test.py
test_getitem (__main__.Test) ... ok
test_reversed (__main__.Test) ... ok
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK