unittest中用ddt传入参数时,传入的参数有几个元素,用例就会执行几次
下面是一个简单的演示,可以复制后直接运行
import numpy as np
import unittest
import ddt
@ddt.ddt # 使用了ddt装饰器的方法
class Demo(unittest.TestCase):
@ddt.data(1, 2) # 迭代的参数值可以直接是一组数
def test_case_1(self, v): # 迭代的参数个数需要与方法中的形参个数一致
print(f"case_1运行结果v:{v}")
@ddt.data([4,5,6]) # 迭代的参数值可以是一个列表
def test_case_2(self, v): # 整个列表是一个参数,用例仅运行了1次
print(f"case_2运行结果v:{v}")
list=np.arange(1,4).tolist() #生成一个从1到3的列表[1,2,3]
@ddt.data(*list) # 迭代的参数值是一个列表中的多个独立元素
def test_case_3(self, v): # 列表有几个元素,用例执行几次
print(f"case_3运行结果v:{v}")
@ddt.data((1, 2), [3, 4]) # 迭代的参数值类型可以为元组或列表
@ddt.unpack # 当迭代的参数为多维数组时,需要使用该装饰器来解析参数
def test_case_4(self, v1, v2):
print(f"case_4运行结果v1:{v1} v2:{v2}")
@ddt.data({"v3": 1, "v4": 2}, {"v3": 3, "v4": 4}) # 迭代的参数值类型可以为字典,字典的key值需要与形参的名称一致
@ddt.unpack
def test_case_5(self, v3, v4):
print(f"case_5运行结果v3:{v3} v4:{v4}")
if __name__ == '__main__':
unittest.main()
运行结果