讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在Python测试中捕获自定义Warning警告

大芳姑娘_2097

大芳姑娘_2097

发布时间:2026-09-30 08:48:16

|

839人浏览过

|

来源于php中文网

原创

应使用 warnings.catch_warnings(record=True) 或 pytest.warns 捕获自定义 Warning;需确保警告类继承 Warning、设 record=True 或 simplefilter("always"),且 pytest 配置未忽略警告。

如何在python测试中捕获自定义warning警告

用 warnings.catch_warnings 临时捕获自定义 Warning

Python 的 warnings 模块默认不抛出异常,即使你触发了自定义警告类,测试里也看不到它——除非显式捕获。最直接的方式是用 warnings.catch_warnings(record=True) 上下文管理器,它能拦截所有警告并存为列表。

注意:必须设 record=True,否则 warnings 对象为空;还要确保你的自定义警告类继承自 Warning(不能是 Exception 或普通 object)。

  • 示例中触发 MyCustomWarning 后,w 是 warnings.WarningMessage 实例,其 .category 是警告类,.message 是实际内容
  • 若想匹配部分消息文本,可用 str(w.message) 做 in 判断,避免硬编码全等
  • 不加 record=True 时,warnings 被丢弃,列表为空,容易误判“没警告”
import warnings
<p>class MyCustomWarning(Warning):
pass</p><p>def risky_func():
warnings.warn("something's off", MyCustomWarning)</p><p>def test_risky_func_warns():
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")  # 必须设为 always,否则可能被忽略
risky_func()
assert len(w) == 1
assert issubclass(w[0].category, MyCustomWarning)
assert "something's off" in str(w[0].message)

用 pytest.warns 断言更简洁

如果你用 pytest,pytest.warns 是专为测试警告设计的上下文管理器,语法更紧凑,失败时提示也更清晰。它内部也是调用 warnings.catch_warnings,但封装了常见逻辑。

关键点在于:它只捕获指定类型的警告(或其子类),且自动验证是否真的发生了——如果没触发,测试直接失败。

Python Use Agent
Python Use Agent

智能执行Python任务,自动生成、执行代码并反馈结果,无需额外配置,兼容旧命令。

下载

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

  • 支持传入具体警告类,如 pytest.warns(MyCustomWarning)
  • 也可传字符串匹配消息,如 pytest.warns(MyCustomWarning, match="off")
  • 若警告被全局过滤(例如 filterwarnings = ignore 在 pyproject.toml 中),pytest.warns 会失效,需检查 pytest 配置
  • 不支持同时断言多个不同警告类型,一次只能盯一种
import pytest
<p>def test_with_pytest_warns():
with pytest.warns(MyCustomWarning, match="off"):
risky_func()

为什么 assertRaises 对 Warning 无效

unittest.TestCase.assertRaises 只捕获 Exception 及其子类,而 Warning 继承自 BaseException 但不是 Exception,所以它根本不会被 assertRaises 捕到——代码会静默通过,造成假阳性。

  • 错误写法:self.assertRaises(MyCustomWarning, risky_func) → 永远不报错,但也没捕获到警告
  • 本质原因:Python 把 Warning 设计为“非中断式提示”,和 Exception 的语义完全不同
  • 如果真想让警告变成异常(仅调试用),可临时设 warnings.simplefilter("error"),但这会改变行为,不适合常规测试逻辑

自定义 Warning 类定义的常见疏漏

自定义警告类本身看似简单,但两个细节常被忽略,导致测试始终不匹配:

  • 必须显式继承 Warning(或其子类如 UserWarning),写成 class MyWarning: 或 class MyWarning(Exception): 都不行
  • 模块路径影响匹配:如果测试里 import 的是 mylib.MyCustomWarning,但触发警告时用的是 from mylib import MyCustomWarning 再 warn(..., MyCustomWarning),通常没问题;但如果跨文件重定义同名类(比如在测试文件里又写了一遍),issubclass 会返回 False
  • 警告类别在 w.category 中是类对象,不是字符串,别用 w.category == "MyCustomWarning" 这种比较

测试里捕获 Warning 看似简单,但真正卡住人的往往是 warning 类型没对上、record=True 忘加、或者 pytest 配置把警告吞掉了。动手前先确认三件事:类继承链对不对、上下文里 filter 设没设对、测试运行时有没有外部 warning filter 干扰。

热门AI工具

更多
豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

相关专题

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

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

1611

2023.07.20

python能做什么
python能做什么

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

3904

2023.07.25

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

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

1609

2023.07.31

python教程
python教程

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

22517

2023.08.03

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

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

2767

2023.08.04

python eval
python eval

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

2807

2023.08.04

scratch和python区别
scratch和python区别

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

1123

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

0

2026.09.30

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn