掌握Python自动化测试利器（pytest测试框架从零入门到实战）

一、为什么选择 pytest？

相比 Python 自带的 unittest 框架，pytest测试框架 具有以下优势：

• 语法简洁，无需继承特定类
• 自动发现测试用例（以 test_ 开头的函数或文件）
• 强大的断言机制，直接使用 assert
• 丰富的插件生态（如 HTML 报告、覆盖率分析等）
• 支持参数化测试、fixture 等高级功能

二、安装 pytest

确保你已安装 Python（建议 3.6+），然后通过 pip 安装 pytest：

pip install pytest

安装完成后，可通过以下命令验证是否成功：

pytest --version

三、编写第一个测试用例

创建一个名为 test_example.py 的文件，输入以下内容：

def add(a, b):    return a + bdef test_add():    assert add(2, 3) == 5    assert add(-1, 1) == 0

在终端运行：

pytest test_example.py

如果看到绿色的 PASSED，恭喜你！你已经成功运行了第一个 Python自动化测试 用例。

四、使用 fixture 管理测试依赖

fixture 是 pytest 的核心特性之一，用于设置测试前的环境（如数据库连接、临时文件等），并在测试后自动清理。

import pytest@pytest.fixturedef sample_data():    return [1, 2, 3, 4, 5]def test_sum(sample_data):    assert sum(sample_data) == 15def test_length(sample_data):    assert len(sample_data) == 5

这里 sample_data 是一个 fixture，被两个测试函数共享，避免了重复代码。

五、参数化测试：一次写多组数据

使用 @pytest.mark.parametrize 可以对同一逻辑进行多组数据测试：

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

这相当于写了 4 个独立的测试用例，但代码更简洁。

六、生成 HTML 测试报告（可选进阶）

安装 pytest-html 插件：

pip install pytest-html

运行测试并生成报告：

pytest --html=report.html --self-contained-html

打开生成的 report.html，即可查看美观的可视化测试结果。

七、总结

通过本教程，你已经掌握了 pytest测试框架 的基本用法，包括编写测试函数、使用 fixture、参数化测试等核心技能。这些知识足以支撑你在日常开发中编写高质量的 单元测试教程 所涵盖的基础场景。

记住，良好的测试习惯能极大提升代码的健壮性和可维护性。现在就开始为你的 Python 项目添加 pytest 测试吧！

关键词回顾：pytest测试框架、Python自动化测试、单元测试教程、pytest入门指南