Python unittest 简单实现参数化的方法

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 简单实现参数化的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。