Python中unittest采用不同的參數組合產生獨立的test case

我們在使用Python的unittest做自動化或者單元測試時，有時需要一個測試用例根據不同的輸入、輸出組合而執行多次，但是，unittest中一個用例只能有一組參數組合執行，如果采用循環的方式，在生成的測試報告中也只會有一個測試用例的統計，單實際上我們需要一組參數就統計為一個測試用例，又或者將用例復制N份（這種方式誰然可以滿足需求，單實際上維護太麻煩，如果有一個改動，需要復制N份），那么我們應該采用什么樣的辦法來實現我們的需求呢？經過查找資料和百度，我們可以采用Python的setattr函數在運行時產生test_開頭的測試用例方法，腳本如下：

import unittest
from test import test_support
class MyTestCase(unittest.TestCase):
    def setUp(self):
        pass

    def clear(self):
        pass

    def getTest(self, arg1, arg2):#定義的函數，最終生成的測試用例的執行方法
        pass

    @staticmethod   
    def getTestFunc(arg1, arg2):
        def func(self):
            self.getTest(arg1, arg2)
        return func
        
def __generateTestCases():
    arglists = [('arg11', 'arg12'), ('arg21', 'arg22'), ('arg31', 'arg32')]
    for args in arglists:
        setattr(MyTestCase, 'test_func_%s_%s'%(args[0], args[1]),MyTestCase.getTestFunc(*args))#通過setattr自動為TestCase類添加成員方法，方法以“test_func_”開頭

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

到此，我們完成了所想要的，在此基礎上可以增加參數化等，來實現自動化測試的要求。