pip install paramunittest
import unittest import paramunittest # 参数化两组元祖类型数据 @paramunittest.parametrized( (8,5), (34,20) ) class UTestDemo(paramunittest.ParametrizedTestCase): def setParameters(self, numa,numb): # 把参数化数据传递到测试类中 self.numa=numa self.numb=numb def test_num(self): print('numa=%d,numb=%d'%(self.numa,self.numb)) self.assertGreater(self.numa, self.numb) if __name__ == '__main__': unittest.main()
执行以上代码,测试结果如下
import unittest import paramunittest # 参数化两组列表类型数据 @paramunittest.parametrized( [8,5], [34,20] ) class UTestDemo(paramunittest.ParametrizedTestCase): def setParameters(self, numa,numb): # 把参数化数据传递到测试类中 self.numa=numa self.numb=numb def test_num(self): print('numa=%d,numb=%d'%(self.numa,self.numb)) self.assertGreater(self.numa, self.numb) if __name__ == '__main__': unittest.main()
测试结果如下
import unittest import paramunittest # 参数化两组字典类型数据 @paramunittest.parametrized( {'numa':8,'numb':5}, {'numa':34,'numb':20} ) class UTestDemo(paramunittest.ParametrizedTestCase): def setParameters(self, numa,numb): # 把参数化数据传递到测试类中 self.numa=numa self.numb=numb def test_num(self): print('numa=%d,numb=%d'%(self.numa,self.numb)) self.assertGreater(self.numa, self.numb) if __name__ == '__main__': unittest.main()
执行结果
通过以上示例可以看出,元祖、字典、列表等都可以实现参数化,参数还可通过函数或者数据对象传入
#参数通过数据对象传入进去 testdata=[{'numa':8,'numb':5},{'numa':34,'numb':20}] @paramunittest.parametrized( *testdata )
# 参数通过函数传入进去 def get_data(): return [{'numa':8,'numb':5},{'numa':34,'numb':20}] @paramunittest.parametrized( *get_data() )