从3.4版开始,Python
supports a simple subtest syntax when writing unittests.一个简单的例子可能如下所示:
import unittest class NumbersTest(unittest.TestCase): def test_successful(self): """A test with subtests that will all succeed.""" for i in range(0,6): with self.subTest(i=i): self.assertEqual(i,i) if __name__ == '__main__': unittest.main()
运行测试时,输出将是
python3 test_foo.py --verbose test_successful (__main__.NumbersTest) A test with subtests that will all succeed. ... ok ---------------------------------------------------------------------- Ran 1 test in 0.000s OK
但是,在我的实际用例中,子测试将依赖于更复杂的迭代,并检查每个子测试的非常不同的东西.因此,我宁愿让每个子测试计数并在输出中列出一个单独的测试用例(在本例中为Ran 6测试)以获得完整的图像.
这在Python中使用plain unittest模块是否可行? The nose test generator feature将单独输出每个测试,但如果可能的话,我希望与标准库保持兼容.
解决方法
你可以继承unittest.TestResult:
class NumbersTestResult(unittest.TestResult): def addSubTest(self,test,subtest,outcome): # handle failures calling base class super(NumbersTestResult,self).addSubTest(test,outcome) # add to total number of tests run self.testsRun += 1
然后在NumbersTest中覆盖run函数:
def run(self,test_result=None): return super(NumbersTest,self).run(NumbersTestResult())
对不起,我现在无法在完全正常工作的环境中对此进行测试,但这应该可以解决问题.