doctest 是python自带的模块, 通过在注释里写代码,以及结果来进行简单的测试
def square(x):
"""Squares x.
>>> square(2)
4
>>> square(-2)
4
>>> square(5)
21
"""
return x * x
doctest.testmod()
**********************************************************************
File "__main__", line 7, in __main__.square
Failed example:
square(5)
Expected:
21
Got:
25
**********************************************************************
1 items had failures:
1 of 3 in __main__.square
***Test Failed*** 1 failures.
Out[19]:
TestResults(failed=1, attempted=3)
注意>>>
后面接测试代码, 中间有空格,下一行是结果