Python數據驅動(ddt)

import unittest
import ddt  #第三方庫

data=[[1,2],[3,4],[5,6]]

@ddt.ddt

class MyTestCase(unittest.TestCase):

    #只有一個參數時
    @ddt.data(1,2,3)
    def test_01(self,a):
        print(a)

    @ddt.data(*data)#表示可參數，若傳參是data,則后面的取值a=[[1,2],[3,4],[5,6]]
    @ddt.unpack
    def test_02(self,a,b):
        print(a,'----',b)

    @ddt.data([1,2],[3,4])#和上面的相似，這里未使用變量
    @ddt.unpack
    def test_03(self,a,b):
        print(a, '----', b)



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