【python单元测试unittest】加载测试套的代码修正

【python单元测试unittest】原创地址：点击打开链接

好文章，简单易懂，收藏了。

最后一处加载测试套的代码有2处有错误。看到有人在问。所以写了这篇博文。

源码更正如下：

import unittest
import re
import os
import sys

def regressionTest():
path = os.path.abspath(os.path.dirname(sys.argv[0]))
files = os.listdir(path)
test = re.compile("test\.py", re.IGNORECASE)
#test = re.compile("test[a-z]*.py", re.IGNORECASE)
files = filter(test.search, files)
filenameToModuleName = lambda f: os.path.splitext(f)[0]
moduleNames = map(filenameToModuleName, files)
modules = map(__import__, moduleNames)

load = unittest.defaultTestLoader.loadTestsFromModule
return unittest.TestSuite(map(load, modules))

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

并且代码的含义与文章说明不一致。其本意是以test.py结尾的测试用例文件。

正则表达式可以根据需要进行修改。