第十六章：开发工具-doctest:通过文档完成测试-运行测试-由模块运行

16.2.7 运行测试
前面的例子都使用doctest内置的命令行测试运行工具。测试单个模块时，这很容易也很方便，不过随着包划分到多个文件，这很快会变得很麻烦。对于这些情况，很多其他方法会更高效。

16.2.7.1 由模块运行
可以在模块最下面包含相应指令来对源代码运行doctest。

def my_function(a,b):
    """
    >>> my_function(2,3)
    6
    >>> my_function('a',3)
    'aaa'
    """
    return a * b

if __name__ == '__main__':
    import doctest
    doctest.testmod()

只有当前模块名是__main__时才会调用testmod()，这可以确保仅当模块作为主程序调用时才会运行测试。

testmod()的第一个参数是一个模块，包含需要扫描的代码(检查其中是否有测试)。其他测试脚本可以使用这个特性导入实际代码，并依次运行各个模块中的测试。

import doctest_simple

if __name__ == '__main__':
    import doctest
    doctest.testmod(doctest_simple)

通过导入各个模块并运行它们的测试，可以为工程构造一个测试套件。