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)

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

1 个答案: