Python pytest断言怎么写

在Python的pytest框架中，断言是通过使用内置的assert语句来编写的。当你在测试函数中使用assert语句时，如果断言失败，pytest会捕获这个失败，并将其报告为一个失败的测试用例。

下面是一个简单的例子，展示了如何在pytest中使用断言：

# test_example.py

def add(a, b):
    return a + b

def test_addition():
    # 使用assert进行断言
    assert add(2, 3) == 5
    assert add(5, -2) == 3
    # 如果下面的断言失败，测试将会失败
    assert add(1, 2) == 4  # 这里会失败，因为1+2不等于4

在上面的例子中，test_addition函数包含了几个断言，用于检查add函数的正确性。如果所有的断言都通过了，那么测试将会成功。如果任何一个断言失败了，pytest将会报告一个失败的测试，并提供失败的原因。

pytest还提供了一些额外的功能来增强断言的可读性和调试体验：

1. assert语句可以包含一个可选的错误消息，当断言失败时，这个消息将会被显示出来。

assert add(1, 2) == 4, "1 + 2 应该等于 3"

2. pytest提供了assert语句的扩展版本，称为assert表达式，它允许你在不使用==的情况下进行比较，并且会自动提供更详细的失败信息。

assert add(1, 2) is 3, "1 + 2 应该等于 3"

3. 如果你想要在断言失败时得到更多的上下文信息，你可以使用pytest的raises断言来检查异常是否被正确抛出。

def divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero.")
    return a / b

def test_division_by_zero():
    with pytest.raises(ValueError) as exc_info:
        divide(1, 0)
    assert "Cannot divide by zero." in str(exc_info.value)

在这个例子中，test_division_by_zero函数使用pytest.raises来断言divide函数在尝试除以零时是否会抛出一个ValueError异常，并且检查异常消息是否正确。

使用这些功能，你可以编写出更加健壮和易于调试的测试用例。