前言:之前记录了一次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)