在Python中编写测试函数：unittest和doctest

在Python中，有两个常用的模块可以用于编写测试函数: unittest和doctest。

1. unittest模块：

unittest模块是Python内置的一个单元测试框架，它提供了一系列的断言方法和测试装置，可以方便地进行单元测试。

使用unittest模块编写测试函数的一般步骤如下：

1) 首先，创建一个继承自unittest.TestCase的测试类，这个类中可以包含多个测试函数。

2) 在测试类中，每个测试函数都应该以"test_"开头，并且不带任何参数。

3) 在测试函数中，使用unittest模块提供的断言方法来判断结果是否符合预期。例如，assertEqual断言方法用于判断两个值是否相等，assertTrue断言方法用于判断一个条件是否为True等等。

4) 最后，使用unittest模块提供的一些测试装置来运行测试函数，获取测试结果，并输出测试报告。例如，使用unittest.main()函数来运行测试类中的所有测试函数，并且输出测试结果。

下面是一个使用unittest模块编写的测试函数的例子：

import unittest

def add(x, y):
    return x + y

class MyTestCase(unittest.TestCase):

    def test_add(self):
        result = add(2, 3)
        self.assertEqual(result, 5)

if __name__ == '__main__':
    unittest.main()

2. doctest模块：

doctest模块是Python标准库中的一个模块，它允许您通过编写文档字符串的方式来编写测试函数。doctest模块会自动提取文档字符串中的示例代码，并且执行这些代码，然后与预期结果进行比较，从而完成测试。

使用doctest模块编写测试函数的一般步骤如下：

1) 首先，创建一个包含了测试函数的.py文件。

2) 在测试函数的文档字符串中，编写示例代码以及预期结果。

3) 最后，使用doctest模块提供的runtest函数来运行测试函数，获取测试结果，并输出测试报告。

下面是一个使用doctest模块编写的测试函数的例子：

def add(x, y):
    """
    返回两个数的和。

    >>> add(2, 3)
    5
    >>> add(4, 5)
    9
    >>> add(10, 20)
    30
    """
    return x + y

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

总结：

unittest模块是一种基于类的测试框架，需要创建测试类和测试函数，使用断言方法进行断言，然后运行测试类来获取测试结果。

doctest模块是一种基于文档字符串的测试框架，可以直接在文档字符串中编写示例代码和预期结果，然后运行测试函数来获取测试结果。