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免费学习笔记(深入)”;

万物追踪
万物追踪

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的用法、作用、函数功能相关内容,阅读专题下面的文章了解更多详细教程。

2

2026.02.03

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

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

82

2025.09.18

python 全局变量
python 全局变量

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

98

2025.09.18

抖音网页版入口与视频观看指南 抖音官网视频在线访问
抖音网页版入口与视频观看指南 抖音官网视频在线访问

本专题汇总了抖音网页版的入口链接、官方登录页面以及视频观看入口,帮助用户快速访问抖音网页版,提供免登录访问方式和直接进入视频播放页面的方法,确保顺利浏览和观看抖音视频。

19

2026.02.04

学习通网页版入口与在线学习指南 学习通官网登录与使用方法
学习通网页版入口与在线学习指南 学习通官网登录与使用方法

本专题详细汇总了学习通网页版入口与登录方法,提供学习通官方网页端入口、学生登录平台、网页版使用指南等内容,帮助用户快速稳定地登录学习通官网,顺利进入学习平台,提升学习效率和体验。

6

2026.02.04

Python Web 框架 Django 深度开发
Python Web 框架 Django 深度开发

本专题系统讲解 Python Django 框架的核心功能与进阶开发技巧,包括 Django 项目结构、数据库模型与迁移、视图与模板渲染、表单与认证管理、RESTful API 开发、Django 中间件与缓存优化、部署与性能调优。通过实战案例,帮助学习者掌握 使用 Django 快速构建功能全面的 Web 应用与全栈开发能力。

2

2026.02.04

Java 流式处理与 Apache Kafka 实战
Java 流式处理与 Apache Kafka 实战

本专题专注讲解 Java 在流式数据处理与消息队列系统中的应用,系统讲解 Apache Kafka 的基础概念、生产者与消费者模型、Kafka Streams 与 KSQL 流式处理框架、实时数据分析与监控,结合实际业务场景,帮助开发者构建 高吞吐量、低延迟的实时数据流管道,实现高效的数据流转与处理。

2

2026.02.04

Golang 容器化与 Docker 实战
Golang 容器化与 Docker 实战

本专题深入讲解 Golang 应用的容器化与 Docker 部署,涵盖 Docker 基础概念、容器构建与镜像管理、Go 应用的 Dockerfile 编写、跨平台容器部署与优化、Docker Compose 和 Kubernetes 部署工具。通过实际案例,帮助学习者掌握 如何将 Golang 应用容器化并实现高效部署与管理,提升系统的可扩展性与运维效率。

3

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号