Python的unittest做参数化测试

约定

参数化case的名字必须以 "param_" 为前缀，后面跟真正的test名字；数据提供函数必须是classmethod，以 "collection_" 为前缀，后面跟真正的test名字。

比如 parameterized_test_add 和 collection_test_add 就是一组参数化case，其中testcase基础名字为test_add，参数化后具体的case为test_add_0, test_add_1, test_add_2 等等。为实现此功能，必须重载unittest的 TestCase 和 TestLoader。

重载unittest.TestCase

class Test(unittest.TestCase):    def __init__(self, methodName='runTest'):        def isParameterizedMethod(attrname):            return attrname.startswith("param") and \                hasattr(getattr(self, attrname), '__call__')        testFnNames = filter(isParameterizedMethod, dir(self))        for func in testFnNames:            name = func.split("_", 1)[1]            collect = "collection_" + name            if hasattr(getattr(self, collect), '__call__'):                collectFunc = getattr(self, collect)                array = collectFunc()                for index in xrange(len(array)):                    test = "%s_%d" % (name, index)                    setattr(self.__class__, test, getattr(self, func)(array[index]))        # must called at last        unittest.TestCase.__init__(self, methodName)

重载unittest.TestLoader

class Loader(unittest.TestLoader):    def getTestCaseNames(self, testCaseClass):        """Return a sorted sequence of method names found within testCaseClass        """        testFnNames = unittest.TestLoader.getTestCaseNames(self, testCaseClass)        def isParameterizedMethod(attrname, testCaseClass=testCaseClass,                         prefix="parameterized"):            return attrname.startswith(prefix) and \                hasattr(getattr(testCaseClass, attrname), '__call__')        testFnNames0 = filter(isParameterizedMethod, dir(testCaseClass))        for func in testFnNames0:            name = func.split("_", 1)[1]            collect = "collection_" + name            if hasattr(getattr(testCaseClass, collect), '__call__'):                collectFunc = getattr(testCaseClass, collect)                for item in xrange(len(collectFunc())):                    testFnNames.append("%s_%d" % (name, item))        if self.sortTestMethodsUsing:            testFnNames.sort(key=_CmpToKey(self.sortTestMethodsUsing))        return testFnNames

编写测试用例

from unittest import *from Test import *from Loader import *class TestFunctions(Test):    @classmethod    def collection_test_add(cls):        return [1,2,3,5]    def parameterized_test_add(self, x):        def test_body(self):            print(x * x)        return test_bodyif __name__ == '__main__':    suite = Loader().loadTestsFromTestCase(TestFunctions)    runner = unittest.TextTestRunner()    rc = runner.run(suite)    print(rc)

在该用例中，真正的testcase定义在test_body函数中。collection_test_add 必须是一个无参的classmethod，返回一个list；parame_test_add 必须为非 classmethod 的成员函数，接受一个入参，该入参为 collection_test_add 所返回的 list 的元素，显然，该 list 的元素可以是任意数据类型，可以是list，tuple，dict等等，这样在test_body内可以接收更加丰富的输入。

本例中，collection_test_add 所返回的 list 中有4个元素，依次生成 test_add_0, test_add_1, test_add_2, test_add_3共4个具体的case。