0

0

如何在 Python 单元测试中正确 Mock 直接导入的函数

花韻仙語

花韻仙語

发布时间:2026-02-04 16:28:03

|

942人浏览过

|

来源于php中文网

原创

如何在 Python 单元测试中正确 Mock 直接导入的函数

当函数以 `from module import func` 方式被导入时,需在**使用该函数的模块作用域内**进行 patch(如 `@patch("entry.read_sql")`),而非原始定义模块,否则 mock 不生效。

在 Python 单元测试(尤其是使用 unittest.mock.patch)中,mock 的目标必须是对象在运行时实际被引用的位置,而非其原始定义位置。这是初学者最容易混淆的关键点。

以你的代码为例:

# entry.py
from sql_services import read_sql  # ← 此处将 read_sql 绑定为 entry 模块的局部名称

def foo():
    df = read_sql("SELECT * FROM schema.table")  # ← 实际调用的是 entry.read_sql!
    return df

虽然 read_sql 定义在 sql_services 中,但 foo() 内部访问的是 entry 模块的全局变量 read_sql —— 这是 Python 导入机制决定的:from ... import 会在当前命名空间创建一个新引用。因此,mock 必须指向 "entry.read_sql",而不是 "sql_services.read_sql"。

✅ 正确的测试写法如下:

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

Firebase Studio
Firebase Studio

谷歌推出的AI全栈开发平台

下载
# test_entry.py
import pytest
import unittest.mock as mock
import pandas as pd

from entry import foo

@mock.patch("entry.read_sql")  # ← 关键:patch 使用方模块中的名称
def test_do_a_read(mock_read_sql):
    mock_read_sql.return_value = pd.DataFrame({"id": [1, 2]})

    result = foo()  # ← 此时调用的是被 mock 的 entry.read_sql

    assert len(result) == 2
    mock_read_sql.assert_called_once_with("SELECT * FROM schema.table")

⚠️ 注意事项:

  • ❌ @patch("sql_services.read_sql") 对 foo() 无效,因为 foo 并未通过 sql_services.read_sql 访问该函数;
  • ✅ Patch 路径必须与 导入后实际使用的全限定名一致(即 module_name.object_name);
  • 若 entry.py 中存在多个导入(如 from sql_services import read_sql, write_sql),需分别 patch entry.read_sql 和 entry.write_sql;
  • 使用 pytest 时,也可改用 monkeypatch fixture 实现更灵活的局部 mock(适合复杂场景),但 @patch 仍是标准、清晰的首选。

? 小技巧:可通过 print(entry.read_sql) 或 assert entry.read_sql is not sql_services.read_sql 验证导入是否创建了独立引用,加深理解。

总结:Mock 的本质是“替换运行时符号表中的对象”,不是“替换源码定义”。始终 patch 被测试代码所在模块中该符号的名称,即可避免因导入方式不同导致的 mock 失效问题。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

187

2023.09.27

python print用法与作用
python print用法与作用

本专题整合了python print的用法、作用、函数功能相关内容,阅读专题下面的文章了解更多详细教程。

4

2026.02.03

全局变量怎么定义
全局变量怎么定义

本专题整合了全局变量相关内容,阅读专题下面的文章了解更多详细内容。

82

2025.09.18

python 全局变量
python 全局变量

本专题整合了python中全局变量定义相关教程,阅读专题下面的文章了解更多详细内容。

98

2025.09.18

Python 数据库优化与性能调优
Python 数据库优化与性能调优

本专题专注讲解 Python 在数据库性能优化中的应用,包括数据库连接池管理、SQL 查询优化、索引设计与使用、数据库事务管理、分布式数据库与缓存系统的结合。通过分析常见性能瓶颈,帮助开发者掌握 如何优化数据库操作,提升 Python 项目在数据库层的响应速度与处理能力。

0

2026.02.05

Java 微服务与 Spring Cloud 实战
Java 微服务与 Spring Cloud 实战

本专题讲解 Java 微服务架构的开发与实践,重点使用 Spring Cloud 实现服务注册与发现、负载均衡、熔断与限流、分布式配置管理、API Gateway 和消息队列。通过实际项目案例,帮助开发者理解 如何将传统单体应用拆分为高可用、可扩展的微服务架构,并有效管理和调度分布式系统中的各个组件。

0

2026.02.05

C++ 多线程编程与线程池设计
C++ 多线程编程与线程池设计

本专题深入讲解 C++ 中的多线程编程与线程池设计,涵盖 C++11/14/17 的线程库、线程同步机制(mutex、condition_variable、atomic)、线程池设计模式、任务调度与优化、并发瓶颈分析与解决方案。通过多个实际案例,帮助开发者掌握 如何设计高效的线程池管理系统,提升 C++ 程序在高并发场景下的性能与稳定性。

0

2026.02.05

微信网页版文件传输助手教程合集
微信网页版文件传输助手教程合集

本专题整合了微信网页版文件传输助手教程、入口等等内容,阅读专题下面的文章了解更多详细内容。

15

2026.02.04

微信文件过期恢复教程
微信文件过期恢复教程

本专题整合了微信文件过期恢复方法、技巧教程,阅读专题下面的文章了解更多详细内容。

10

2026.02.04

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.4万人学习

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

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