dt 是第三方模块,需安装, pip install ddt
DDT包含类的装饰器ddt和两个方法装饰器data(直接输入测试数据)
通常情况下,data中的数据按照一个参数传递给测试用例,如果data中含有多个数据,以元组,列表,字典等数据,需要自行在脚本中对数据进行分解或者使用unpack分解数据。
- from ddt import ddt,data,unpack
- import unittest
- test_data1 = [{"username": "zhangsan", "pwd": "zhangsan"},
- {"username": "lisi", "pwd": "lisi"},
- {"username": "wangwu", "pwd": "wangwu"},
- ]
- test_data2 = [{"username": "wukong", "pwd": "wukong"},
- {"username": "wuneng", "pwd": "woneng"},
- {"username": "wujing", "pwd": "wujing"},
- ]
- @ddt
- class Test(unittest.TestCase):
- def setUp(self):
- print("===Start!===")
- def tearDown(self):
- print("===end!===")
- @data(*test_data1)
- def test_ddt1(self, data):
- print(data)
- ```
- test_ddt1的测试结果是OK的,因为test_data2作为一个整体传给了data,所以value打印的值为test_data1
- ```
- @data(*test_data2)
- def test_ddt2(self, data):
- print(data['username'])
- ```
- test_ddt2的测试结果是OK的,因为test_data2作为一个整体传给了data,然后根据字典取出value值
- ```
- @data([3,2,1],[5,3,2],[10,4,6])
- @unpack #@unpack,那么[3,2,1]被分解开,按照用例中的三个参数传递
- def test_minus(self,a,b,expected):
- actual = int(a)-int(b)
- expected = int(expected)
- print(actual,expected)
- self.assertEqual(actual, expected)
- ```
- test_minus的测试结果也是ok的,由于在 @ data(...)下加了 @ unpack, 代表会把数据分解,得到3组测试数据
- ```
- @data([2, 3], [4, 5]) #没有@unpack,那么[2,3]当成一个参数传入用例运行
- def test_compare(self, a, b):
- print(a,b)
- self.assertEqual(a, b)
- ```
- test_compare的测试结果是fail的,由于没有加 @ unpack, 虽然还是会被理解成2组测试数据,但是[2, 3]
- 作为一个整体被传给了a, 因为b就没有值传入了,所以一执行后报了
- TypeError: test_compare() missing 1 required positional argument: 'b' 这句错
- ```
- if __name__ == "__main__":
- unittest.main()
联系客服