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

Python中如何模拟导入模块进行单元测试_使用unittest.mock库

梦浩姑娘_9812

梦浩姑娘_9812

发布时间:2026-05-18 13:22:26

|

365人浏览过

|

来源于php中文网

原创

mock.patch 必须在测试函数执行前生效,因为它仅拦截后续的导入或属性访问;若模块已提前导入,则无法替换原始对象。

python中如何模拟导入模块进行单元测试_使用unittest.mock库

mock.patch 为什么必须在测试函数执行前生效

因为 patch 实际上是在目标模块被导入、代码被执行前,临时替换成 Mock 对象。如果模块已在测试前被导入(比如写在文件顶部),patch 就无法拦截原始导入行为——它只作用于后续的 import 或属性访问。

常见错误现象:ModuleNotFoundError 没出现,但 mock 没生效,被测函数仍调用真实模块。

  • 正确做法:在 test_* 函数内部使用 with patch("module.submodule.function") as mock_func:
  • 或用装饰器:在测试函数上加 @patch("requests.get"),注意装饰器顺序(越靠近函数的越先执行)
  • 绝对路径必须从被测代码「看到的导入路径」出发,不是你本地文件路径。例如被测代码写了 from requests import get,要 patch "mymodule.get";如果写了 import requests,则 patch "mymodule.requests.get"

mock.return_value 和 mock.side_effect 的关键区别

return_value 是静态返回,适合模拟固定响应;side_effect 支持抛异常、返回不同值、甚至调用真实函数,更贴近真实交互逻辑。

使用场景:需要测试 HTTP 请求失败、数据库查询空结果、多次调用返回不同数据时,side_effect 不可替代。

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

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • mock_func.return_value = {"status": "ok"} → 每次调用都返回这个 dict
  • mock_func.side_effect = [ValueError("timeout"), {"data": 42}] → 第一次抛异常,第二次返回 dict
  • mock_func.side_effect = lambda url: {"url": url, "cached": True} → 支持参数感知的动态返回

如何安全地 mock 全局变量或类实例方法

全局变量(如配置字典、连接池单例)和实例方法容易因 patch 位置不当导致状态污染,影响其他测试用例。

性能与兼容性影响:对类方法 patch 过深(如 patch("module.Class.method"))可能绕过 __init__ 或属性初始化逻辑,导致断言失败但原因难查。

  • 优先 patch 被测模块中「使用该变量/方法的位置」,而不是定义位置。例如模块 A 定义了 DB_POOL,模块 B 导入并使用了它,测试 B 时应 patch "B.DB_POOL"
  • mock 类实例方法时,确保 patch 的是绑定后的方法(即实例调用路径),而非类定义本身,否则 self 可能为 None
  • 用 autospec=True 参数可自动继承原函数签名,防止调用时参数错位却不报错

AssertionError 显示 “Expected call not found” 怎么快速定位

这不是 mock 失效,而是调用记录和预期不匹配——可能参数类型不对、调用次数少一次、或者用了 assert_called_once() 但实际调用了两次(包括隐式调用)。

容易踩的坑:字符串和 bytes 混用、字典 key 顺序(Python 3.7+ 有序但测试中别依赖)、浮点数直接比较。

  • 用 mock_func.assert_called_with("https://api.com", timeout=5) 前,先打印 mock_func.call_args 看实际传了什么
  • 检查是否误用了 assert_called_once():它等价于 assert_called() and call_count == 1,而有些库会内部调用多次(如 logging 配置)
  • 避免在 patch 中用 new_callable=MagicMock 后又手动设 return_value,这会覆盖 autospec 行为,让参数校验失效
mock 最难的不是写法,是判断“该在哪儿 patch”和“该 mock 到哪一层”。多看被测函数里 import 和调用的实际路径,比背文档更管用。

热门AI工具

更多
音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

豆包大模型

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

WorkBuddy

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

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

Flowith
Flowith Hot

Flowith是一款基于画布和智能体工作流的 AI 协作工作空间。

DeepSeek

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

相关专题

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

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

1851

2023.07.20

python能做什么
python能做什么

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

5024

2023.07.25

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

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

1809

2023.07.31

python教程
python教程

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

30237

2023.08.03

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

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

3607

2023.08.04

python eval
python eval

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

3607

2023.08.04

scratch和python区别
scratch和python区别

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

1243

2023.08.11

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

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

616

2023.08.10

Tailwind CSS 如何制作导航栏和卡片
Tailwind CSS 如何制作导航栏和卡片

用 Tailwind CSS 制作导航栏和卡片,可先搭建语义化结构,再组合 Flex、Grid、间距、圆角与阴影类。这里整理横向导航、移动端菜单、卡片网格及悬停效果的相关文章,帮助你把常见页面组件做得清晰且适配不同屏幕。

0

2026.11.03

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

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

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