Python数据驱动ddt

import ddt
import unittest

"""
ddt模块包含了一个类的装饰器ddt和两个方法的装饰器：

data：包含多个你想要传给测试用例的参数；

file_data:会从json或yaml中加载数据；

通常data中包含的每一个值都会作为一个单独的参数传给测试方法，如果这些值是用元组或者列表传进来的，可以用unpack方法将其自动分解成多个参数。
"""
test_data2 = [{"a": "1", "pwd": "123456"},
              {"b": "2", "pwd": "123456"}]

@ddt.ddt
class Test(unittest.TestCase):
    def setUp(self):
        print("Start!")

    def tearDown(self):
        print("end!")

    @ddt.file_data('url.yaml')
    def test_ddt1(self, **data):
        print(data)

    """
    执行2次，yaml文件中是一个含有2个元素的list
    Start!
    {'url': 'http://feature.v3.0.1.xxxx.com/albums',
     'headers': {'Authorization': 'x3JpemF0aW9ucyIsIxxwz8V1gn80yE-b8', 
     'Accept': 'application/vnd.xxx.admin+json'},
      'Body': {'url': 'http://list.youku.com/show/ixxxbf.html?spm=a2h0j.11185381.bpmodule-playpage-righttitle.5~H2~A'},
       'code': {'status_code': 200, 'message': 'success'}}
    end!
    Start!
    {'url': 'http://feature.v3.0.1.xxxx.com/albums',
     'headers': {'Authorization': 'x3JpemF0aW9ucyIsIxxwz8V1gn80yE-b8',
      'Accept': 'application/vnd.xxx.admin+json'},
       'Body': {'url': 'http://list.youku.com/show/ixxxbf.html?spm=a2h0j.11185381.bpmodule-playpage-righttitle.5~H2~A'},
        'code': {'status_code': 200, 'message': 'success'}}
    end!
    """

    @ddt.data(*test_data2)
    def test_ddt2(self, data):
        print(data)
        return data
    """
    Start!
    {'a': '1', 'pwd': '123456'}
    end!
    Start!
    {'b': '2', 'pwd': '123456'}
    end!
    """


# suit = unittest.TestSuite()
# suit.addTest(Test('test_ddt1'))
# suit.addTest(unittest.makeSuite(Test))
# unittest.TextTestRunner().run(suit)

if __name__ == "__main__":
    unittest.main()