python中test_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:UsersmengmaDesktop1.py", line 25, in __main__.User

Failed example:

u.name

Expected:

'fkjava'

Got:

'fkit'

**********************************************************************

File "C:UsersmengmaDesktop1.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:UsersmengmaDesktop1.py", line 8, in __main__.square

Failed example:

square(3)

Expected:

9

Got:

6

**********************************************************************

File "C:UsersmengmaDesktop1.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() 函数会自动提取该模块的说明文档中的测试用例，并执行这些测试用例，最终生成测试报告。如果存在没有通过的测试用例，程序就会显示有多少个测试用例没有通过；如果所有测试用例都能通过测试，则不生成任何输出结果。