pytest

适用场景

当用户希望执行以下操作时，使用此技能：

• 使用 pytest 编写并运行 Python 单元测试
• 使用 fixture 进行测试设置、清理和依赖注入
• 参数化测试，以使用多组输入运行测试
• 使用 pytest-mock 或 unittest.mock 模拟依赖项
• 通过 pytest.ini、pyproject.toml 或 conftest.py 配置 pytest
• 使用 pytest-cov 生成覆盖率报告
• 使用 pytest-xdist 并行运行测试

如何使用此技能

工作流

1. 编写测试函数，名称以 test_ 开头，或编写 Test 类
2. 在 conftest.py 中使用 fixture 进行共享设置和依赖注入
3. 使用 pytest CLI 运行测试，并根据需要应用标记和筛选条件
4. 使用覆盖率和报告插件分析结果

1. 基本测试

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

def test_string_upper():
    assert "hello".upper() == "HELLO"

2. Fixture

import pytest

@pytest.fixture
def sample_user():
    return {"name": "Alice", "email": "[email protected]"}

def test_user_name(sample_user):
    assert sample_user["name"] == "Alice"

@pytest.fixture(autouse=True)
def reset_database(db):
    db.clear()
    yield
    db.clear()

3. 参数化

import pytest

@pytest.mark.parametrize("input,expected", [
    ("hello", 5),
    ("", 0),
    ("pytest", 6),
])
def test_string_length(input, expected):
    assert len(input) == expected

4. 模拟

def test_api_call(mocker):
    mock_get = mocker.patch("requests.get")
    mock_get.return_value.json.return_value = {"status": "ok"}

    result = fetch_status()
    assert result == "ok"
    mock_get.assert_called_once()

5. 运行测试

# Run all tests
pytest

# Verbose with pattern filter
pytest -v -k "test_user"

# With coverage
pytest --cov=mypackage --cov-report=html

# Parallel execution
pytest -n auto

最佳实践

• 保持测试相互独立，不共享可变状态；使用 fixture 进行设置和清理
• 使用清晰、具体的测试名称，说明预期行为
• 避免依赖测试执行顺序；使用 conftest.py 定义共享 fixture
• 在 CI 中运行完整测试套件；使用 --cov 检查覆盖率，并使用 --maxfail=1 尽早失败
• 使用标记（@pytest.mark.slow）对测试分类并运行部分测试

资源

• 官方文档：https://docs.pytest.org/
• pytest-cov：https://pytest-cov.readthedocs.io/
• pytest-xdist：https://github.com/pytest-dev/pytest-xdist

关键词

pytest、Python 测试、fixture、参数化、模拟、conftest、覆盖率、pytest-cov、pytest-xdist、单元测试、标记