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

Python 测试框架：unittest 与 pytest 详解

1. 测试概述

在软件开发中，测试是确保代码质量和功能正确性的重要环节。通过测试，可以验证代码是否符合预期，并且在后续的开发过程中，避免引入新的错误。下面将介绍使用 unittest 和 pytest 这两个 Python 测试框架进行测试的方法。

2. 使用 unittest 进行测试

2.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.2 使用 setUp 和 tearDown 的测试

接下来，我们构建一个更复杂的测试，涉及随机数生成器。以下是随机数生成器的代码：

class MyRand(object):
    def set(self, p1, p2, x0, modulus):
        self.__p1 = p1
        self.__p2 = p2
        self.__x