pytest有哪些高级特性

pytest 是一个功能强大且灵活的 Python 测试框架，具有许多高级特性，可以帮助开发者编写更高效、可维护的测试代码。以下是一些 pytest 的高级特性：

1. 参数化测试

通过 @pytest.mark.parametrize 装饰器，可以轻松地对同一个测试函数进行多次调用，每次使用不同的参数组合。

import pytest

@pytest.mark.parametrize("input, expected", [
    (1, 2),
    (2, 3),
    (3, 4),
])
def test_increment(input, expected):
    assert input + 1 == expected

2. Fixtures

Fixtures 是 pytest 中用于设置和清理测试环境的机制。它们可以用于管理数据库连接、文件操作等资源。

import pytest

@pytest.fixture
def db_connection():
    # 设置数据库连接
    conn = create_connection()
    yield conn
    # 清理数据库连接
    conn.close()

def test_query(db_connection):
    result = db_connection.execute("SELECT * FROM users")
    assert len(result) > 0

3. Markers

Markers 允许你为测试函数添加自定义标记，以便在运行测试时进行筛选。

import pytest

@pytest.mark.slow
def test_long_running():
    # 模拟长时间运行的测试
    pass

@pytest.mark.smoke
def test_smoke():
    # 模拟冒烟测试
    pass

你可以使用 -m 选项来筛选带有特定标记的测试：

pytest -m slow

4. Plugins

pytest 支持丰富的插件生态系统，可以通过安装插件来扩展其功能。例如，pytest-django 插件可以方便地进行 Django 应用的测试。

pip install pytest-django

然后在测试文件中导入并使用插件：

import pytest

@pytest.fixture(scope="session", autouse=True)
def django_db_setup(django_db_blocker):
    with django_db_blocker.unblock():
        # 设置 Django 数据库
        pass

5. 自定义命令行选项

你可以通过 addoption 方法为 pytest 添加自定义命令行选项，并在测试中使用这些选项。

def pytest_addoption(parser):
    parser.addoption("--env", action="store", default="dev", help="environment to run tests against")

@pytest.fixture(scope="session")
def env(request):
    return request.config.getoption("--env")

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

def test_environment(env):
    assert env in ["dev", "prod"]

6. 并行测试

pytest-xdist 插件允许你并行运行测试，从而加快测试速度。

pip install pytest-xdist

然后使用 -n 选项指定并行运行的进程数：

pytest -n 4

7. 自定义报告

pytest-html 插件可以生成 HTML 格式的测试报告，便于查看和分析测试结果。

pip install pytest-html

然后在命令行中运行测试并生成报告：

pytest --html=report.html

8. Mocking

pytest-mock 插件提供了方便的 mocking 功能，可以在测试中模拟外部依赖。

pip install pytest-mock

然后在测试中使用 mocker fixture：

def test_external_dependency(mocker):
    mock = mocker.patch("module.ExternalDependency")
    mock.return_value.some_method.return_value = "mocked result"
    result = module.SomeClass().some_method()
    assert result == "mocked result"

这些高级特性使得 pytest 成为一个非常强大且灵活的测试框架，能够满足各种复杂的测试需求。