Python自动化测试：pytest框架入门指南

要从零开始搭建基于 pytest 的测试框架，请按照以下步骤操作：1. 安装 pytest 并创建符合命名规范的测试文件；2. 编写测试函数并使用 pytest 执行测试，通过 -v 查看详细结果；3. 组织测试结构，将测试代码放入 tests/ 目录并按模块划分；4. 使用 fixture 管理前置/后置操作，通过 conftest.py 共享常用 fixture；5. 安装 pytest-html 和 pytest-xdist 插件生成 HTML 报告并支持多进程执行；6. 将测试框架集成到 CI 工具中实现持续集成。掌握这些步骤即可搭建高效的自动化测试体系，但关键仍在于设计和维护高质量的测试用例。

自动化测试是提升开发效率和代码质量的重要手段，而 Python 中的 pytest 框架凭借其简洁、灵活和强大的插件生态，成为很多团队首选的测试工具。如果你想用 Python 做自动化测试，从零开始搭建基于 pytest 的测试框架其实并不难。

安装与基本使用

首先确保你已经安装了 Python，然后通过 pip 安装 pytest：

pip install pytest

安装完成后，你可以创建一个以 test_ 开头或结尾的 Python 文件（比如 test_sample.py），在里面写一些简单的测试函数：

def test_addition():
    assert 1 + 1 == 2

def test_failure():
    assert 1 + 1 == 3

运行测试只需要在命令行输入：

pytest

它会自动查找所有符合命名规范的测试文件并执行。失败的测试会被标记出来，方便你快速定位问题。

• 成功的测试显示为 .
• 失败的测试显示为 F
• 可以加 -v 参数查看详细输出

组织测试结构

随着测试数量增加，你需要合理组织测试文件和目录结构。一个常见的做法是把测试代码放在单独的 tests/ 目录下，并按照功能模块划分子目录或文件。

例如：

project_root/
├── src/
│   └── my_module.py
└── tests/
    ├── test_module_a.py
    ├── test_module_b.py
    └── __init__.py

这样不仅便于维护，也方便 CI 工具集成。你还可以使用 pytest 的 fixture 功能来复用初始化逻辑，比如准备数据库连接、登录状态等。

使用 fixture 管理前置/后置操作

fixture 是 pytest 最实用的功能之一，可以用来替代 unittest 的 setUp 和 tearDown。

比如你想在每个测试前都登录一次系统：

import pytest

@pytest.fixture
def login():
    print("Logging in...")
    token = "abc123"
    yield token
    print("Logout")

def test_profile(login):
    assert login == "abc123"

上面的例子中，login 函数是一个 fixture，在测试函数中作为参数传入时会被自动调用。yield 前面的部分相当于 setup，后面部分相当于 teardown。

你可以将常用的 fixture 放在 conftest.py 文件中，这样整个项目都可以共享它们。

集成报告与持续集成

光跑测试还不够，我们还需要生成测试报告。pytest 提供了很多插件支持，其中最常用的是 pytest-html 和 pytest-xdist。

安装 html 报告插件：

pip install pytest-html

生成 HTML 报告的命令如下：

pytest --html=report.html

如果你希望测试跑得更快，还可以用 pytest-xdist 插件实现多进程并发执行：

pip install pytest-xdist
pytest -n 4  # 使用4个进程

这些报告和并行执行能力可以很方便地集成到 Jenkins、GitHub Actions 等 CI 工具中，实现每日构建和自动化回归测试。

基本上就这些。掌握好 pytest 的基本结构、fixture 使用和报告生成，就可以开始搭建属于自己的自动化测试体系了。不复杂但容易忽略的是测试用例的设计和维护，这才是真正决定自动化价值的地方。