如何使用pytest进行单元测试

pytest 是一个功能强大且易于使用的 Python 单元测试框架。以下是如何使用 pytest 进行单元测试的步骤：

1. 安装 pytest

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

pip install pytest

2. 编写测试用例

假设你有一个简单的函数需要测试，比如一个计算两个数之和的函数：

# math_utils.py

def add(a, b):
    return a + b

接下来，你需要编写一个测试文件来测试这个函数。测试文件的命名通常以 test_ 开头，例如 test_math_utils.py。

# test_math_utils.py

from math_utils import add

def test_add():
    assert add(2, 3) == 5
    assert add(-1, 1) == 0
    assert add(0, 0) == 0

3. 运行测试

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

pytest

pytest 会自动发现并运行所有符合命名规范的测试文件和测试函数，并输出测试结果。

4. 查看测试报告

pytest 默认会提供一个简洁的文本报告。如果你想看到更详细的报告，可以使用 -v 选项：

pytest -v

5. 使用 Fixtures

pytest 提供了强大的 fixtures 功能，可以用来设置和清理测试环境。例如，你可以创建一个 fixture 来提供测试数据：

# conftest.py

import pytest

@pytest.fixture
def sample_data():
    return [1, 2, 3, 4, 5]

然后在测试函数中使用这个 fixture：

# test_math_utils.py

from math_utils import add
import pytest

def test_add_with_fixture(sample_data):
    for data in sample_data:
        assert add(data, data) == data * 2

6. 参数化测试

你可以使用 @pytest.mark.parametrize 装饰器来参数化测试函数，从而用不同的输入数据运行相同的测试逻辑：

# test_math_utils.py

import pytest
from math_utils import add

@pytest.mark.parametrize("a, b, expected", [
    (2, 3, 5),
    (-1, 1, 0),
    (0, 0, 0),
    (10, -5, 5),
])
def test_add_parametrized(a, b, expected):
    assert add(a, b) == expected

7. 断言

pytest 支持多种断言方式，包括 assert 语句、pytest.raises 用于测试异常等。例如：

# test_math_utils.py

from math_utils import divide

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

通过这些步骤，你可以使用 pytest 进行有效的单元测试。pytest 还有许多其他高级功能和插件，可以根据需要进行探索和使用。