# 16.7. 全部放在一起 你已经学习了足够的知识，现在来分析本章样例代码的前七行：读取一个目录并从中导入选定的模块。 ## 例 16.16. `regressionTest` 函数 ``` def regressionTest(): path = os.path.abspath(os.path.dirname(sys.argv[0])) files = os.listdir(path) 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)) ``` 让我们一行行交互地看。假定当前目录是 `c:\diveintopython\py`，其中有包含本章脚本在内的本书众多样例。正如在 [第 16.2 节 “找到路径”](finding_the_path.html "16.2. 找到路径") 中所见，脚本目录将存于 `path` 变量，因此让我们从这里开始以实打实的代码起步。 ## 例 16.17. 步骤 1：获得所有文件 ``` >>> import sys, os, re, unittest >>> path = r'c:\diveintopython\py' >>> files = os.listdir(path) >>> files ['BaseHTMLProcessor.py', 'LICENSE.txt', 'apihelper.py', 'apihelpertest.py', 'argecho.py', 'autosize.py', 'builddialectexamples.py', 'dialect.py', 'fileinfo.py', 'fullpath.py', 'kgptest.py', 'makerealworddoc.py', 'odbchelper.py', 'odbchelpertest.py', 'parsephone.py', 'piglatin.py', 'plural.py', 'pluraltest.py', 'pyfontify.py', 'regression.py', 'roman.py', 'romantest.py', 'uncurly.py', 'unicode2koi8r.py', 'urllister.py', 'kgp', 'plural', 'roman', 'colorize.py'] ``` | | | | --- | --- | | \[1\] | `files` 是由脚本所在目录的所有文件和目录构成的列表。(如果你已经运行了其中的一些样例，可能还会看到一些 `.pyc` 文件。) | ## 例 16.18. 步骤 2：找到你关注的多个文件 ``` >>> test = re.compile("test\.py$", re.IGNORECASE) >>> files = filter(test.search, files) >>> files ['apihelpertest.py', 'kgptest.py', 'odbchelpertest.py', 'pluraltest.py', 'romantest.py'] ``` | | | | --- | --- | | \[1\] | 这个正则表达式将匹配以 `test.py` 结尾的任意字符串。注意，你必须转义这个点号，因为正则表达式中的点号通常意味着 “匹配任意单字符”，但是你实际上想匹配的事一个真正的点号。 | | \[2\] | 被编译的正则表达式就像一个函数，因此你可以用它来过滤文件和目录构成的大列表，找寻符合正则表达式的所有元素。 | | \[3\] | 剩下的是一个单元测试脚本列表，因为只有它们是形如 `SOMETHINGtest.py` 的文件。 | ## 例 16.19. 步骤 3：映射文件名到模块名 ``` >>> filenameToModuleName = lambda f: os.path.splitext(f)[0] >>> filenameToModuleName('romantest.py') 'romantest' >>> filenameToModuleName('odchelpertest.py') 'odbchelpertest' >>> moduleNames = map(filenameToModuleName, files) >>> moduleNames ['apihelpertest', 'kgptest', 'odbchelpertest', 'pluraltest', 'romantest'] ``` | | | | --- | --- | | \[1\] | 正如你在 [第 4.7 节 “使用 lambda 函数”](../power_of_introspection/lambda_functions.html "4.7. 使用 lambda 函数") 中所见，`lambda` 快餐式地创建内联单行函数。这里应用你在 [例 6.17 “分割路径名”](../file_handling/os_module.html#splittingpathnames.example "例 6.17. 分割路径名") 中已经见过的，标准库的 `os.path.splitext` 将一个带有扩展名的文件名返回成只包含文件名称的那部分。 | | \[2\] | `filenameToModuleName` 是一个函数。`lambda` 函数并不比你以 `def` 语句定义的普通函数神奇。你可以如其他函数一样地调用 `filenameToModuleName`，它也将如你所愿：从参数中剔除扩展名。 | | \[3\] | 现在你可以通过 `map` 把这个函数应用于单元测试文件列表中的每一个文件。 | | \[4\] | 结果当然如你所愿：以指代模块的字符串构成的一个列表。 | ## 例 16.20. 步骤 4：映射模块名到模块 ``` >>> modules = map(__import__, moduleNames) >>> modules [<module 'apihelpertest' from 'apihelpertest.py'>, <module 'kgptest' from 'kgptest.py'>, <module 'odbchelpertest' from 'odbchelpertest.py'>, <module 'pluraltest' from 'pluraltest.py'>, <module 'romantest' from 'romantest.py'>] >>> modules[-1] <module 'romantest' from 'romantest.py'> ``` | | | | --- | --- | | \[1\] | 正如你在 [第 16.6 节 “动态导入模块”](dynamic_import.html "16.6. 动态导入模块") 中所见，你可以通过 `map` 和 `__import__` 的协同工作，将模块名 (字符串) 映射到实际的模块 (像其他模块一样可以被调用和使用)。 | | \[2\] | `modules` 现在是一个模块列表，其中的模块和其他模块一样。 | | \[3\] | 该列表的最后一个模块_是_ `romantest` 模块，和通过 `import romantest` 导入的模块完全等价。 | ## 例 16.21. 步骤 5：将模块载入测试套件 ``` >>> load = unittest.defaultTestLoader.loadTestsFromModule >>> map(load, modules) [<unittest.TestSuite tests=[ <unittest.TestSuite tests=[<apihelpertest.BadInput testMethod=testNoObject>]>, <unittest.TestSuite tests=[<apihelpertest.KnownValues testMethod=testApiHelper>]>, <unittest.TestSuite tests=[ <apihelpertest.ParamChecks testMethod=testCollapse>, <apihelpertest.ParamChecks testMethod=testSpacing>]>, ... ] ] >>> unittest.TestSuite(map(load, modules)) ``` | | | | --- | --- | | \[1\] | 模块对象的存在，使你不但可以像其他模块一样地使用它们；通过类的实例化和函数的调用，你还可以内省模块，从而弄清楚已经有了那些类和函数。这正是 `loadTestsFromModule` 方法的工作：内省每一个模块并为每个模块返回一个 `unittest.TestSuite` 对象。每个 `TestSuite` (测试套件) 对象都包含一个 `TestCase` 对象的列表，每个对象对应着你的模块中的一个测试方法。 | | \[2\] | 最后，你将`TestSuite`列表封装成一个更大的测试套件。`unittest` 模块会很自如地遍历嵌套于测试套件中的树状结构，最后深入到独立测试方法，一个个加以运行并判断通过或是失败。 | 自省过程是 `unittest` 模块经常为我们做的一项工作。还记得我们的独立测试模块仅仅调用了看似神奇的 `unittest.main()` 函数就大刀阔斧地完成了全部工作吗？`unittest.main()` 实际上创建了一个 `unittest.TestProgram` 的实例，而这个实例实际上创建了一个 `unittest.defaultTestLoader` 的实例并以调用它的模块启动它。 (如果你不给出，如何知道调用它的模块是哪一个？通过使用同样神奇的 `__import__('__main__')` 命令，动态导入正在运行的模块。我可以就 `unittest` 模块中使用的所有技巧和技术写一本书，但那样我就没法写完这本了。) ## 例 16.22. 步骤 6：告知 `unittest` 使用你的测试套件 ``` if __name__ == "__main__": unittest.main(defaultTest="regressionTest") ``` | | | | --- | --- | | \[1\] | 在不使用 `unittest` 模块来为我们做这一切的神奇工作的情况下，你实际上已自己做到了。你已经创建了一个自己就能导入模块、调用 `unittest.defaultTestLoader` 并封装于一个测试套件的 `regressionTest` 函数。现在你所要做的不是去寻找测试并以通用的方法构建一个测试套件，而是告诉 `unittest` 前面那些，它将调用 `regressionTest` 函数，而它会返回可以直接使用的 `TestSuite`。 |