【Python】使用pytest进行异常断言的终极指南

最新推荐文章于 2024-12-30 09:38:57 发布

civilpy

最新推荐文章于 2024-12-30 09:38:57 发布

阅读量557

点赞数 3

文章标签： python mysql 数据库

本文链接：https://blog.csdn.net/baidu_22713341/article/details/139644919

版权

基本原理

在Python的自动化测试中，pytest 是一个非常流行的测试框架，它提供了一种简单的方式来编写和执行测试用例。当我们想要验证一个函数或方法在特定条件下是否抛出了预期的异常时，pytest 提供了几种断言方法来帮助我们进行验证。

代码示例

以下是几种使用 pytest 进行异常断言的方法，以及相应的示例代码。

示例1：使用 `pytest.raises`

这是最基本的断言异常的方式。我们使用 pytest.raises 来断言一个代码块抛出了特定的异常。

import pytest

def test_divide_by_zero():
    with pytest.raises(ZeroDivisionError):
        1 / 0

示例2：捕获异常信息

除了断言异常类型，我们还可以捕获异常信息，进一步验证异常的详细信息。

def test_value_error():
    with pytest.raises(ValueError) as exc_info:
        raise ValueError("Invalid value")
    assert str(exc_info.value) == "Invalid value"

示例3：断言多个异常类型

有时候，我们可能不确定会抛出哪种异常，但我们知道它会是几种类型之一。

def test_multiple_exceptions():
    with pytest.raises(Exception) as exc_info:
        raise RuntimeError("Something went wrong")
    assert isinstance(exc_info.value, (RuntimeError, TypeError))

注意事项

使用上下文管理器：pytest.raises 需要作为上下文管理器使用，即使用 with 语句。
异常类型要准确：断言的异常类型应该是你期望抛出的确切类型，或者其父类。
异常信息验证：如果需要验证异常信息，可以使用 as exc_info 来捕获异常，并使用 exc_info.value 来访问异常实例。

结论

通过使用 pytest 的异常断言功能，我们可以确保我们的代码在遇到错误情况时能够如预期那样抛出异常。这不仅有助于我们验证错误处理逻辑的正确性，也是编写健壮代码的重要一环。掌握这些基本的断言方法，可以让我们在进行单元测试时更加得心应手。

![](https://img-blog.csdnimg.cn/direct/08a02446429a4a85bb9963d696b9f5bd.png)
>
> 【痕迹】QQ+微信朋友圈和聊天记录分析工具1.0.4 （1）纯Python语言实现，使用Flask后端，本地分析，不上传个人数据。
>
> （2）内含QQ、微信聊天记录保存到本地的方法，真正实现自己数据自己管理。
>
> （3）数据可视化分析QQ、微信聊天记录，提取某一天的聊天记录与大模型对话。
> 
> **下载地址：https://www.alipan.com/s/x6fqXe1jVg1**
>