Pytest教程：详解Pytest的三种多级断言方法

当涉及到测试代码时，多级断言可以提供更全面的测试覆盖，并且允许开发者一次性检查多个方面的代码行为。在 Pytest 中，有三种主要的方式来实现多级断言：使用多个普通的 assert 语句、使用 pytest-assume 插件以及使用 pytest-check 插件。本文将详细介绍这三种方法，并提供源码示例以及实战场景。

1. 使用多个普通的 assert 语句

使用多个普通的 assert 语句是最直接的多级断言方法。在一个测试函数中，可以使用多个 assert 语句来检查不同的断言条件，即使其中一个失败也不会中断整个测试函数的执行。

python 复制代码

def test_multiple_assertions():
    data = [1, 2, 3, 4, 5]
    
    assert len(data) == 5
    assert data[0] == 1
    assert data[3] == 4
    assert sum(data) == 15

2. 使用 pytest-assume 插件

pytest-assume 插件允许我们在一个测试函数中对多个条件进行假设，并且只有当所有假设都通过时才会继续执行后续断言。

python 复制代码

import pytest

def test_multiple_assumptions():
    data = [1, 2, 3, 4, 5]
    
    pytest.assume(len(data) == 5)
    pytest.assume(data[0] == 1)

    # 如果前两个假设通过才会继续执行以下断言
    assert data[3] == 4
    assert sum(data) == 15

安装

pip install pytest-assume

使用

1.导入 pytest-assume 模块

在测试文件中导入 assume 模块：

python 复制代码

from pytest_assume import assume

2.编写测试函数

编写您的测试函数，并在其中使用 assume 来执行断言。示例：

python 复制代码

def test_some_functionality():
    result = some_function()
    assume(result == expected_result_1, "First assertion failed")
    assume(another_condition(result), "Second assertion failed")
    assume(yet_another_condition(result), "Third assertion failed")

3.运行测试

使用 pytest 命令来运行测试：

pytest <your_test_file.py>

3. 使用 pytest-check 插件

pytest-check 插件提供了一种更简洁的方式来实现多级断言，通过使用 check 装饰器和 check 对象，可以在一个测试函数中进行多个条件的检查。

python 复制代码

import pytest
from pytest_check import check

def test_using_check_decorator():
    data = [1, 2, 3, 4, 5]
    
    with check:
        assert len(data) == 5
        assert data[0] == 1
        assert data[3] == 4
        assert sum(data) == 15

安装

使用以下命令安装 pytest-check：

pip install pytest-check

使用

1.导入 pytest-check 模块

在测试文件中导入 check 模块：

python 复制代码

from pytest_check import check

2.编写参数化测试函数

使用 @check.parametrize 装饰器来定义参数化测试函数。示例：

python 复制代码

@check.parametrize('input_data, expected_output', [
    (1, 2),
    (3, 6),
    (5, 10),
])
def test_multiply_by_two(input_data, expected_output):
    result = input_data * 2
    check.equal(result, expected_output, f"Failed for input: {input_data}")

3.运行测试

使用 pytest 命令来运行测试：

pytest <your_test_file.py>

4.三种断言对比

1. assert

使用方式：assert 是 Python 的内置断言语句，可以直接在测试函数中使用。

特点：当一个 assert 失败时，后续的代码将不会执行，并且测试函数将会失败。

适用场景：适用于简单的断言检查，一旦有一个断言失败就立即终止测试。

2. pytest-assume

使用方式 ：pytest-assume 是一个 Pytest 的插件，通过 pytest.assume 可以设置假设条件。

特点：所有的假设语句都会被执行，只有在最后的 assert 失败时才会导致测试函数失败。

适用场景：适用于需要对多个条件进行假设，但希望能够继续执行所有断言的情况。

3. pytest-check

使用方式 ：pytest-check 是另一个 Pytest 的插件，通过 check 装饰器和 check 对象可以实现多级断言。

特点：所有的 check 内的断言都会被执行，如果有任何一个断言失败，会将失败信息收集并一起展示。

适用场景：适用于需要做多个断言检查，但希望能够查看所有断言的结果而不是立即终止测试。

总结

assert 适用于简单的断言检查，一旦失败就终止测试。

pytest-assume 适用于对多个条件进行假设，并在最后统一验证。

pytest-check 适用于多级断言，收集所有断言结果并一起展示。