Python unittest 理论上是不建议参数驱动的,其用例应该专注单元测试,确保每个method的逻辑正确。
引用Stack Overflow的一个答案,
“单元测试应该是独立的,没有依赖项的。这确保了每个用例都有非常具体而专一的测试反应。传入参数会破坏单元测试的这个属性,从而使它们在某种意义上无效。使用测试配置是最简单的方法,也是更合适的方法,因为单元测试不应该依赖外部信息来执行测试。那应该集成测试要做的。”
但是实际操作过程中,时不时还是有控制入参的需求的。比如,我想简单实现一个web功能的cross-browser测试……
下面列出一些学习到的解决方案 (ENV: WIN 10, python36)。
利用类的属性
这种方法可以不依赖其他第三方库,而且可以将参数化应用到setUpClass 和setUp方法中。
即可以顺利解决一个web 测试脚本cross-browser的验证的问题。
class TestOdd1(unittest.TestCase): NUMBER=1 def runTest(self): """Assert that the item is odd""" self.assertTrue( self.NUMBER % 2 == 1, "Number should be odd") class TestOdd2(TestOdd1): NUMBER=2 if __name__ == '__main__': unittest.main()
nose + ddt
用nose和ddt可以简单的完成参数控制的test case,实际应用的是python的装饰器(decorator)。
写出来有些类似cucumber gherkin当中的scenario outline。
在实验中,这个方法不适用于setUpClass。
pip install nose pip install ddt
import unittest from ddt import ddt, data @ddt class TestOdd(unittest.TestCase): @data(3, 4, 12, 23) def runTest(self, value): self.assertTrue( self.NUMBER % 2 == 1, "Number should be odd")
执行 nosetests my_test.py ,4个test case被执行。
这个方案还支持从外部文件中加载参数。具体可以参考DDT 官方文档。
重写unittest.TestCase的构造函数
出处:Eli Bendersky的博客
定义类ParametrizedTestCase,使之集成unittest.TestCase并重写其构造函数,增加param这个参数。
import unittest class ParametrizedTestCase(unittest.TestCase): """ TestCase classes that want to be parametrized should inherit from this class. """ def __init__(self, methodName='runTest', param=None): super(ParametrizedTestCase, self).__init__(methodName) self.param = param @staticmethod def parametrize(testcase_klass, param=None): """ Create a suite containing all tests taken from the given subclass, passing them the parameter 'param'. """ testloader = unittest.TestLoader() testnames = testloader.getTestCaseNames(testcase_klass) suite = unittest.TestSuite() for name in testnames: suite.addTest(testcase_klass(name, param=param)) return suite
下面是一个包含我们用例的测试脚本,继承了ParametrizedTestCase,
class TestOne(ParametrizedTestCase): def test_something(self): print 'param =', self.param self.assertEqual(1, 1) def test_something_else(self): self.assertEqual(2, 2)
以参数驱动的方式执行用例
uite = unittest.TestSuite() suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=42)) suite.addTest(ParametrizedTestCase.parametrize(TestOne, param=13)) unittest.TextTestRunner(verbosity=2).run(suite)
将得到如下输出,
test_something (__main__.TestOne) ... param = 42 ok test_something_else (__main__.TestOne) ... ok test_something (__main__.TestOne) ... param = 13 ok test_something_else (__main__.TestOne) ... ok ----------------------------------------------- Ran 4 tests in 0.000s OK
以上这篇Python unittest 简单实现参数化的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。