前言:之前记录了一次ddt模式的简单使用,这次再介绍一下ddt的多数据使用。
需求:有的时候你需要用到的数据是一个整体的可迭代数据,而你每条用例里面只需要这个数据中的一个数据。这时ddt也可以解决。
比如你有{"a":1,"b":2,"c":3},这些数据是你从各个地方读取到的,然后你要用这个给ddt去传值,但是你想执行3次,每次依次传值。
废话不多说,直接上代码
import ddt,unittest dict1 = {"a":1,"b":2,"c":3} list1 = [1,2,3,"a",{"b":"1"}] tuple1 = (1,2,3,4,"a") @ddt.ddt class Test(unittest.TestCase): ## 直接传入 @ddt.data(dict1) def test(self,d1): print(d1) @ddt.data(list1) def test1(self,l1): print(l1) @ddt.data(tuple1) def test2(self,t1): print(t1) if __name__ == '__main__': unittest.main()
结果:
... ---------------------------------------------------------------------- Ran 3 tests in 0.000s OK [1, 2, 3, 'a', {'b': '1'}] (1, 2, 3, 4, 'a') {'a': 1, 'b': 2, 'c': 3} Process finished with exit code 0
上面的代码就是直接把整个的参数全部推进去了。但是现在是需要各个参数里面每个值作为一次参数去执行一下,前面学习过多次执行的方法,如下
import ddt,unittest dict1 = {"a":1,"b":2,"c":3} list1 = [1,2,3,"a",{"b":"1"}] tuple1 = (1,2,3,4,"a") @ddt.ddt class Test(unittest.TestCase): ## 分次传入 @ddt.data({"a":1},{"b":2}) def test(self,d1): print(d1) @ddt.data([1],[2]) def test1(self,l1): print(l1) @ddt.data([1],[2]) def test2(self,t1): print(t1) if __name__ == '__main__': unittest.main()
如上,想要每次都传入一个的话,ddt.data()中用逗号隔开就可以满足分次执行了。返回结果如下:
[1]
[2]
[1]
[2]
{'a': 1}
{'b': 2}
......
----------------------------------------------------------------------
Ran 6 tests in 0.001s
OK
但如果我们的测试用data数据很大呢?这个该怎么半,不能一个一个去手动输入呀
这里可以用*来解决这个问题,如下
import ddt,unittest dict1 = {"a":1,"b":2,"c":3} list1 = [1,2,3,"a",{"b":"1"}] tuple1 = (1,2,3,4,"a") @ddt.ddt class Test(unittest.TestCase): ## 直接传入 @ddt.data(*dict1.items()) def test(self,d1): print(d1) @ddt.data(*list1) def test1(self,l1): print(l1) @ddt.data(*tuple1) def test2(self,t1): print(t1) if __name__ == '__main__': unittest.main()
这样就能解决我们的问题了,返回结果如下:
1
2
3
a
{'b': '1'}
1
2
3
4
a
('a', 1)
('b', 2)
('c', 3)
---疲惫的我,只想在街上找辆法拉利或者兰博基尼去试试头够不够铁