python之ddt模式随记二

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

上面的代码就是直接把整个的参数全部推进去了。但是现在是需要各个参数里面每个值作为一次参数去执行一下，前面学习过多次执行的方法，如下

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

但如果我们的测试用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)

来源：https://www.cnblogs.com/dflblog/p/12012846.html

标签

ddt

unittest

data

python