python-ddt装饰器

代码：

import unittest
from ddt import ddt,data,unpack

test_data = [[1,3],[4,5]]
test_data1 = [[1,3], [6,7]]
test_data2 = [{"name":"haha", "pwd":"123"},
              {"name":"gaga", "pwd":"323"}]

@ddt # 装饰类
class TestMath(unittest.TestCase):

    @data(*test_data)
    @unpack
    def test_print_data(self,item,item1):
        print("test_print_data",item)
        print("test_print_data1",item1)

    @data(test_data1)
    def test_one_data(self,item):
        print("test_one_data",item)

    @data(*test_data2)
    @unpack
    def test_dict_data(self,name,pwd):
        print("name:",name)
        print("pwd:",pwd)

 

执行结果：

name: haha
pwd: 123
name: gaga
pwd: 323

test_one_data [[1, 3], [6, 7]]

test_print_data 1
test_print_data1 3
test_print_data 4
test_print_data1 5