我如何使附加的 pyunit python 脚本搜索子文件夹中的测试

Question

以下 python 脚本搜索并执行当前文件夹中的所有 pyunit 测试。

"""Regression testing framework

This module will search for scripts in the same directory named
XYZtest.py.  Each such script should be a test suite that tests a
module through PyUnit.  (As of Python 2.1, PyUnit is included in
the standard library as 'unittest'.)  This script will aggregate all
found test suites into one big test suite and run them all at once.
"""

import sys, os, re, unittest

def regressionTest():
    path = os.path.abspath(os.path.dirname(sys.argv[0]))
    files = os.listdir(path)
    print 'You are HERE! ' ,os.getcwd()
    test = re.compile( ".test\.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")

我怎样才能得到它并在子文件夹中搜索测试文件。

也许是这样的：

import sys, os, re, unittest, fnmatch

    for root, dirs, files in os.walk(path):
        for filename in fnmatch.filter(files, pattern):
            print os.path.join(root, filename)

Answer 1

更改：

def regressionTest():
   path = os.path.abspath(os.path.dirname(sys.argv[0]))

至：

def regressionTest(somearg):
   path = os.path.abspath(os.path.dirname(somearg))

然后：

for root, dirs, files in os.walk(path):
    regressionTest(root)