0

0

Python pytest 框架中的异常断言

舞夢輝影

舞夢輝影

发布时间:2025-09-19 22:48:01

|

691人浏览过

|

来源于php中文网

原创

使用pytest.raises可精确断言异常类型和错误信息,通过上下文管理器捕获异常,并用match参数验证错误消息是否匹配字符串或正则;结合as excinfo可访问异常实例的属性和类型,确保自定义异常的完整性和上下文正确,从而提升测试的健壮性与代码可靠性。

python pytest 框架中的异常断言

在Python的

pytest
框架中,异常断言是确保代码在特定条件下能正确抛出预期错误的关键机制。它让我们能够验证程序的错误处理逻辑是否健全,而不是仅仅关注正常的执行路径。

当我们需要验证一段代码是否会按预期抛出异常时,

pytest.raises
是一个极其强大且直观的工具。它允许我们捕获并检查异常的类型、消息,甚至是异常对象本身的更多细节。

如何在 pytest 中精确地断言异常类型和错误信息?

在编写测试时,我们往往不只是想知道“有没有抛异常”,更想知道“抛的是不是对的异常,消息对不对”。

pytest.raises
在这方面提供了非常精细的控制。

通常,我们会使用

with pytest.raises(ExpectedExceptionType)
这样的上下文管理器。比如,如果我有一个函数
divide(a, b)
,当
b
为零时它应该抛出
ZeroDivisionError
,我的测试会是这样:

立即学习Python免费学习笔记(深入)”;

def divide(a, b):
    if b == 0:
        raise ZeroDivisionError("除数不能为零")
    return a / b

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

这仅仅验证了异常类型。但如果我想更进一步,确认异常消息也符合预期呢?

pytest.raises
match
参数就派上用场了。它可以接受一个正则表达式或一个字符串,用于匹配异常的详细信息。

import pytest

def divide(a, b):
    if b == 0:
        raise ZeroDivisionError("除数不能为零,请检查输入")
    return a / b

def test_divide_by_zero_with_message():
    # 使用字符串匹配
    with pytest.raises(ZeroDivisionError, match="除数不能为零"):
        divide(1, 0)

    # 也可以使用正则表达式,更灵活
    with pytest.raises(ZeroDivisionError, match=r"除数不能为零.*检查输入"):
        divide(1, 0)

    # 假设有个自定义异常
    class MyCustomError(ValueError):
        pass

    def do_something_risky(value):
        if value < 0:
            raise MyCustomError("负数是不允许的!")

    with pytest.raises(MyCustomError, match="负数是不允许的!"):
        do_something_risky(-5)

这种精确的匹配,尤其是在处理复杂的业务逻辑或外部API错误时,显得尤为重要。它能确保我们的错误信息不仅存在,而且是用户或开发者能够理解、并据此采取行动的。毕竟,一个含糊不清的错误信息,和没有错误信息也差不了多少。

处理预期之外的异常行为,pytest 有哪些高级用法?

有时,我们不仅要断言异常的抛出,还需要在异常抛出后,对异常对象本身进行更深入的检查。

pytest.raises
上下文管理器返回的
ExceptionInfo
对象就提供了这种能力。通过
as excinfo
语法,我们可以捕获到这个对象。

import pytest

class ConfigurationError(Exception):
    def __init__(self, message, config_key=None):
        super().__init__(message)
        self.config_key = config_key

def load_config(settings):
    if "database_url" not in settings:
        raise ConfigurationError("缺少数据库连接配置", config_key="database_url")
    return True

def test_missing_database_config():
    with pytest.raises(ConfigurationError) as excinfo:
        load_config({})

    # 检查异常类型
    assert excinfo.type is ConfigurationError
    # 检查异常消息
    assert "缺少数据库连接配置" in str(excinfo.value)
    # 检查自定义属性
    assert excinfo.value.config_key == "database_url"

    # 甚至可以检查异常的traceback
    # assert "load_config" in str(excinfo.traceback) # 通常不推荐直接断言traceback字符串,但知道有这个能力

这里

excinfo.value
就是实际抛出的异常实例。我们可以访问它的任何属性,包括自定义的属性。这对于那些携带额外上下文信息的异常(比如我们自定义的
ConfigurationError
,它有一个
config_key
属性)来说,是极其有用的。它让我们能验证异常不仅仅是类型正确,其内部状态也符合预期。

再比如,我们可能需要测试一个函数,它可能会因为多种原因抛出异常,但我们只关心其中一种。或者,我们想确保某个函数在特定条件下绝不抛出异常。虽然

pytest.raises
主要是为了断言异常的抛出,但如果一个函数在不应该抛出异常时抛出了,测试自然会失败。

AdMaker AI
AdMaker AI

从0到爆款高转化AI广告生成器

下载

对于那些可能抛出多种异常的场景,我们可以在

pytest.raises
中传入一个元组,来断言多种可能的异常类型:

import pytest

def risky_operation(value):
    if value == 0:
        raise ValueError("零值无效")
    if value < 0:
        raise TypeError("负数类型不符")
    return 1 / value

def test_risky_operation_exceptions():
    # 期望抛出ValueError或TypeError
    with pytest.raises((ValueError, TypeError)):
        risky_operation(0)
    with pytest.raises((ValueError, TypeError)):
        risky_operation(-1)

    # 仍然可以检查具体消息
    with pytest.raises(ValueError, match="零值无效"):
        risky_operation(0)

这种处理方式让我们的测试更加健壮,能够覆盖到更多潜在的错误路径。

为什么在测试中异常断言如此重要,它能避免哪些潜在问题?

异常断言在测试策略中扮演着一个不可或缺的角色。它不仅仅是代码覆盖率的一个数字,更是我们对程序健壮性和可靠性承诺的体现。

首先,它确保了错误处理逻辑的正确性。很多时候,我们写下

try...except
块,但很少真正去验证当异常发生时,我们的
except
块是否真的被执行了,或者它是否处理了正确的异常类型。异常断言就是这块缺失的拼图。它强制我们思考:在什么情况下,我的代码会失败?失败时,它应该如何表现?

其次,异常断言是API契约的有效保障。当你开发一个库或模块时,你可能会约定在输入不合法时抛出特定的异常。例如,一个接受文件路径的函数,如果路径不存在,就应该抛出

FileNotFoundError
。通过测试这些异常,我们实际上是在验证我们的API行为符合预期,为使用者提供清晰、可预测的错误反馈。如果后续有人修改了代码,不小心改变了异常行为,测试会立即发现。

再者,它有助于避免“静默失败”。最糟糕的错误往往是那些不声不响就发生了的。如果一个函数在遇到问题时,不是抛出异常,而是返回一个

None
或者一个空列表,那么调用者可能不会立即意识到问题,导致错误在系统深处才被发现,排查起来异常困难。异常断言迫使我们明确地定义和测试这些失败点,确保问题能够及时、明确地暴露出来。

从个人经验来看,我曾遇到过这样的情况:一个关键的配置加载函数,在配置项缺失时,本应抛出

ConfigurationError
,但由于开发人员的疏忽,它只是默默地返回了一个默认值,导致下游组件在运行时行为异常,且难以追踪源头。如果当时有针对性的异常断言,这个问题会在第一时间被发现,并修复。

所以,异常断言不仅仅是测试的一种手段,它更是防御性编程思想在测试层面的体现。它帮助我们构建更稳定、更可靠、更易于维护的软件系统。它让我们在面对各种不可预见的情况时,能够更加自信地说:“我的代码知道如何应对。”

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

773

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

684

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

765

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

719

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1425

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

570

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

751

2023.08.11

c++ 根号
c++ 根号

本专题整合了c++根号相关教程,阅读专题下面的文章了解更多详细内容。

25

2026.01.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 2.8万人学习

R 教程
R 教程

共45课时 | 5.5万人学习

C# 教程
C# 教程

共94课时 | 7.5万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号