python提供了一个比较简便的、可以直接写在function里面的测试方法。
def sum_naturals(n):
"""Return the sum of the first n natural numbers.
>>> sum_naturals(10)
55
>>> sum_naturals(100)
5050
"""
total, k = 0, 1
while k <= n:
total, k = total + k, k + 1
return total
func注释中的
>>> sum_naturals(10)
55
>>> sum_naturals(100)
5050
可以被当作test case被run
from doctest import run_docstring_examples
run_docstring_examples(sum_naturals, globals(), True)