如何用pytest进行单元测试

使用 pytest 进行单元测试是一个非常流行的选择，因为它简单易用，并且功能强大。以下是如何使用 pytest 进行单元测试的基本步骤：

1. 安装 pytest

首先，你需要安装 pytest。你可以使用 pip 来安装：

pip install pytest

2. 编写测试代码

假设你有一个简单的 Python 函数需要测试，比如一个计算平方的函数：

# math_functions.py

def square(number):
    return number * number

3. 编写测试用例

接下来，你需要编写测试用例。测试用例通常放在一个以 test_ 开头的文件中，例如 test_math_functions.py。

# test_math_functions.py

from math_functions import square

def test_square():
    assert square(2) == 4
    assert square(3) == 9
    assert square(0) == 0
    assert square(-2) == 4

4. 运行测试

在命令行中，导航到包含测试文件的目录，然后运行 pytest 命令：

pytest

pytest 会自动发现并运行所有以 test_ 开头的文件和函数。

5. 查看测试结果

pytest 会输出测试结果，包括通过的测试和失败的测试。如果某个测试失败，pytest 会显示详细的错误信息。

6. 使用 Fixtures

pytest 提供了 fixtures 功能，可以用来设置和清理测试环境。例如，如果你需要在每个测试之前创建一些数据，可以使用 fixture：

# test_math_functions.py

import pytest
from math_functions import square

@pytest.fixture
def setup_data():
    return [2, 3, 0, -2]

def test_square(setup_data):
    for number in setup_data:
        assert square(number) == number * number

在这个例子中，setup_data 是一个 fixture，它会在每个测试函数运行之前被调用，并返回一个数据列表。测试函数 test_square 使用这个数据列表来进行测试。

7. 参数化测试

pytest 还支持参数化测试，可以用不同的输入数据多次运行同一个测试函数。可以使用 @pytest.mark.parametrize 装饰器来实现：

# test_math_functions.py

import pytest
from math_functions import square

@pytest.mark.parametrize("number, expected", [
    (2, 4),
    (3, 9),
    (0, 0),
    (-2, 4),
])
def test_square(number, expected):
    assert square(number) == expected

在这个例子中，test_square 函数会被多次调用，每次使用不同的 number 和 expected 参数。

8. 断言

pytest 支持多种断言方式，包括 assert 语句和 pytest.raises 来测试异常：

# test_math_functions.py

from math_functions import divide

def test_divide():
    assert divide(4, 2) == 2

def test_divide_by_zero():
    with pytest.raises(ZeroDivisionError):
        divide(4, 0)

在这个例子中，test_divide 测试正常的除法操作，而 test_divide_by_zero 测试除以零时是否会抛出 ZeroDivisionError 异常。

通过这些步骤，你可以使用 pytest 进行单元测试，并确保你的代码是正确和可靠的。