Python doctest模块：文档测试

所谓文档测试，指的是通过 doctest 模块运行 Python 源文件的说明文档中的测试用例，从而生成测试报告。

前面在查看 Python 的模块文档时，经常会看到如下信息：

>>>os.path.commonprefix (['/usr/lib','/usr/local/lib'])
'/usr/1'
>>>os.path.commonpath(['/usr/lib', '/usr/local/lib'])
'/usr'

用户完全可以将这些代码拷贝到 Python 的交互式解释器中运行，然后可以看到运行结果与文档中示例代码的输出结果完全一致。

上面的说明就是文档测试的注释，文档测试工具可以提取说明文档中的测试用例，其中“>>>”之后的内容表示测试用例，接下来的一行则代表该测试用例的输出结果。文档测试工具会判断测试用例的运行结果与输出结果是否一致，如果不一致就会显示错误信息。

现在定义一个简单的模块，该模块包含一个函数和一个类，程序为该函数和该类提供了说明文档，该文档中包含了测试用例。程序代码如下：

def square (x):
    '''
    一个用于计算平方的函数

    例如
    >>> square(2)
    4
    >>> square(3)
    9
    >>> square(-3)
    9
    >>> square(0)
    0
    '''
    return x * 2 # ①、故意写错的

class User:
    '''
    定义一个代表用户的类，该类包含如下两个属性：
    name - 代表用户的名字
    age - 代表用户的年龄

    例如
    >>> u = User('fkjava', 9)
    >>> u.name
    'fkjava'
    >>> u.age
    9
    >>> u.say('i love python')
    'fkjava说: i love python'
    '''
    def __init__(self, name, age):
        self.name = 'fkit' # ②、故意写错的
        self.age = age
    def say(self, content):
        return self.name + '说: ' + content
if __name__=='__main__':
    import doctest
    doctest.testmod()

上面第一段注释代码就是程序为 square() 函数提供的测试用例，在文档中一共为该函数提供了 4 个测试用例；第二段注释代码是程序为 User 类提供的测试用例，在文档中一共为该类提供了 3 个测试用例，分别用于测试用户的 name 、age 和 say() 方法。

程序最后几行代码进行了判断，如果是直接使用 python 命令来运行该程序（__name__ 等于 __main__），程序将导入 doctest 模块，并调用该模块的 testmod() 函数。

从上面程序可以看到，Python 为文档测试提供了 doctest 模块，该模块的用法非常简单，程序只要调用该模块的 testmod() 函数即可。

运行上面程序，可以看到如下输出结果：

**********************************************************************
File "C:\Users\mengma\Desktop\1.py", line 25, in __main__.User
Failed example:
    u.name
Expected:
    'fkjava'
Got:
    'fkit'
**********************************************************************
File "C:\Users\mengma\Desktop\1.py", line 29, in __main__.User
Failed example:
    u.say('i love python')
Expected:
    'fkjava说: i love python'
Got:
    'fkit说: i love python'
**********************************************************************
File "C:\Users\mengma\Desktop\1.py", line 8, in __main__.square
Failed example:
    square(3)
Expected:
    9
Got:
    6
**********************************************************************
File "C:\Users\mengma\Desktop\1.py", line 10, in __main__.square
Failed example:
    square(-3)
Expected:
    9
Got:
    -6
**********************************************************************
2 items had failures:
   2 of   4 in __main__.User
   2 of   4 in __main__.square
***Test Failed*** 4 failures.

从上面的输出结果可以看出，一共有 4 个测试没有通过，在 User 类中有两个测试没有通过，在 square() 函数中有两个测试没有通过。这是因为上面程序中 ① ② 两行代码故意写错了，其中 ① 号代码用于计算 x 的平方，应该写成 x**2，但漏写了一个星号；② 号代码则应该用传入的 name 参数对 self.name 赋值。

上面显示的测试输出结果也很清晰，每个测试用例结果都包含如下 4 部分：

第一部分：显示在哪个源文件的哪一行。

第二部分：Failed example，显示是哪个测试用例出错了。

第三部分：Expected，显示程序期望的输出结果。也就是在“>>>命令”的下一行给出的运行结果，它就是期望结果。

第四部分：Got，显示程序实际运行产生的输出结果。只有当实际运行产生的输出结果与期望结果一致时，才表明该测试用例通过。

将上面程序中 ① ② 两行代码修改正确，再次使用 python 命令来运行程序，将看不到任何输出结果，说明文档测试中的所有测试用例都通过了。

由此可见，Python 为文档注释提供了 doctest 模块，该模块的用法非常简单，程序只要导入该模块，并调用该模块的 testmod() 函数即可。testmod() 函数会自动提取该模块的说明文档中的测试用例，并执行这些测试用例，最终生成测试报告。如果存在没有通过的测试用例，程序就会显示有多少个测试用例没有通过；如果所有测试用例都能通过测试，则不生成任何输出结果。