【pytest】如何自定义一个断言函数，并在pytest测试中使用它

案例1

在pytest中，你可以通过自定义断言方法来实现更友好的错误输出。这通常涉及到创建一个自定义的断言函数，该函数在断言失败时输出一个清晰、有意义的错误信息。下面是一个简单的例子，说明如何自定义一个断言函数，并在pytest测试中使用它。

首先，你需要创建一个自定义的断言函数。这个函数应该接受你想要测试的值，以及一个预期的值或条件。如果测试失败，函数应该抛出一个AssertionError，并附带一个清晰的错误消息。

def assert_equal_with_message(actual, expected, message):
    """
    自定义断言函数，用于比较两个值是否相等。
    如果不相等，则抛出一个带有自定义错误消息的AssertionError。
    """
    if actual != expected:
        raise AssertionError(message)

然后，在你的pytest测试用例中，你可以使用这个自定义的断言函数来替代标准的assert语句。这样，当断言失败时，你会看到一个更友好的错误消息。

import pytest

def test_example():
    actual_value = 5
    expected_value = 6
    
    # 使用自定义断言函数
    assert_equal_with_message(actual_value, expected_value, f"实际值 {
     actual_value} 不等于预期值