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

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

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

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

2. 使用 pytest-assume 插件

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

import pytestdef test_multiple_assumptions():data = [1, 2, 3, 4, 5]pytest.assume(len(data) == 5)pytest.assume(data[0] == 1)# 如果前两个假设通过才会继续执行以下断言assert data[3] == 4assert sum(data) == 15

安装

pip install pytest-assume

使用

1.导入 pytest-assume 模块

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

from pytest_assume import assume

2.编写测试函数

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

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

3. 使用 pytest-check 插件

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

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

安装

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

pip install pytest-check

使用

1.导入 pytest-check 模块

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

from pytest_check import check

2.编写参数化测试函数

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

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

3.运行测试

使用 pytest 命令来运行测试：

pytest

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适用于多级断言，收集所有断言结果并一起展示。