16、Python 测试框架：unittest 与 pytest 详解

Python 测试框架：unittest 与 pytest 详解

1. 简单测试示例

首先来看一个简单的测试运行示例，运行测试后一切按预期工作：

In [4]: %run test2.py -v
test_f (__main__.TestHailStones) ... 1
2
3
4
5
6
7
8
9
10
ok
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK

2. 使用 setUp 和 tearDown 的测试

之前的测试比较直接，只需包含正确的文件并调用要测试的函数。而这里要构建一个更复杂的测试，在调用测试和函数之前需要进行一些初始化，之后还需要清理。

以随机数生成器为例，使用的简单迭代随机数生成器类代码如下：

class MyRand(object):
    def set(self, p1, p2, x0, modulus):
        self.__p1 = p1
        self.__p2 = p2
        self.__x = x0
        self.__modulus = modulus
    def next(self):
        self.__x = (self.__p1 * self.__x + self.__p2) % self.__modulus
        ret