pytest断言方法封装

在编写测试用例时，我们经常会用到断言来验证代码的正确性。为了使断言的使用更加方便和简洁，我们可以封装一些常用的断言方法。下面是一个示例：

import pytest

def assert_equal(actual, expected):
    assert actual == expected, f"Expected {expected}, but got {actual}"

def assert_not_equal(actual, expected):
    assert actual != expected, f"Expected {expected} to be different from {actual}"

def assert_in(item, container):
    assert item in container, f"{item} not found in {container}"

def assert_not_in(item, container):
    assert item not in container, f"{item} found in {container}"

# 使用示例
def test_addition():
    result = 2 + 2
    assert_equal(result, 4)

def test_subtraction():
    result = 5 - 3
    assert_not_equal(result, 2)

def test_list():
    my_list = [1, 2, 3]
    assert_in(2, my_list)
    assert_not_in(4, my_list)

在上面的示例中，我们封装了四个常用的断言方法：`assert_equal`、`assert_not_equal`、`assert_in`和`assert_not_in`。这些方法接受两个参数：实际值和期望值，并使用 `assert` 语句进行断言判断。如果断言失败，将抛出 `AssertionError` 异常，并输出自定义的错误信息。

通过封装这些断言方法，我们可以在编写测试用例时更加简洁和可读。另外，我们还可以根据需要自定义更多的断言方法来满足不同的测试需求。