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

如何正确地对类实例内部调用的函数进行打桩(Mock)?

酷辰小哥_5201

酷辰小哥_5201

发布时间:2026-01-06 13:58:03

|

836人浏览过

|

来源于php中文网

原创

如何正确地对类实例内部调用的函数进行打桩(Mock)?

本文详解在 python 单元测试中,如何精准 mock 被类方法内部调用的外部函数(如 get_player_account_status),重点解决路径错误、作用域混淆及 side_effect 动态返回等常见问题。

在单元测试中 mock 类内部调用的函数,关键不在于“函数定义在哪”,而在于“它被从哪个模块导入并使用”。你的原始代码中:

# model.py
from get_player_labels import get_player_account_status

这意味着 get_player_account_status 是以 名称 get_player_account_status 被导入到 model 模块的命名空间中的。因此,在测试时必须 patch model.get_player_account_status —— 即该函数在 model 模块中被引用的路径,而非其原始定义路径 get_player_labels.get_player_account_status。

提示词大师-python版
提示词大师-python版

图片提示词生成器?不止如此。 马甲系统 —— 把脑海中的画面,翻译成AI能理解的专业表达。 用得越多,它越懂你:首次需要多问几句确认方向,用久了几乎一说就懂。 用得越多,它越快:缓存机制让后续对话越来越省。 RAG进化:成功案例持续入库,越跑越聪明。 输入「新手指南」查看完整功能介绍

下载

你已尝试 @mock.patch('model.get_player_account_status'),但未生效,根本原因通常是以下之一:

✅ 正确做法:Patch 位置 + 作用域 + 调用时机

  1. 确保 patch 路径绝对准确
    使用 model.get_player_account_status 是正确的,但需确认:

    • model.py 文件在 Python 的 sys.path 中可导入(即 pytest 运行时能 import model);
    • 无拼写错误(如 model 是否实际为 src.model 或 app.model?建议打印 model.__file__ 验证)。
  2. 避免 pytest + unittest 混合导致的 fixture 生命周期冲突
    你当前代码混合了 @pytest.fixture 和 unittest.TestCase,这会导致 setUp() 和 @mock.patch 装饰器行为不可靠(unittest 的 setUp 在 pytest fixture 之后执行,且 @patch 可能未覆盖到 fit() 内部调用)。强烈建议统一为纯 pytest 风格,更简洁、可控:

# test_model.py(推荐重构版)
import pytest
from unittest.mock import patch, MagicMock
from model import PlayerAnomalyDetectionModel

@pytest.fixture
def sample_train_data():
    # 构建示例数据(略)
    return pd.DataFrame({"player": ["p1", "p2", "p3"]})

def test_fit_with_mocked_account_status(sample_train_data):
    # 创建模型实例
    model = PlayerAnomalyDetectionModel()

    # ✅ 正确 patch:mock model 模块中导入的函数
    with patch("model.get_player_account_status") as mock_get_status:
        # 配置 side_effect:按调用顺序返回不同值
        mock_get_status.side_effect = [
            "open", "open", "tosViolation", "tosViolation", "closed"
        ]

        # 执行被测方法
        model.fit(sample_train_data, generate_plots=False)

        # ✅ 断言:验证 mock 被调用(可选)
        assert mock_get_status.call_count == 5
        # 若需检查参数:mock_get_status.assert_any_call("p1", model._account_statuses)
? 为什么 @mock.patch('get_player_labels.get_player_account_status') 失败? 因为 model.py 并未直接调用 get_player_labels.get_player_account_status,而是调用了自己命名空间里的 get_player_account_status。Mock 必须作用于被调用处的命名空间,而非定义处。

✅ side_effect 使用要点

  • side_effect 可接受列表、可调用对象或异常类;
  • 列表形式会按顺序逐次返回,超出长度后抛出 StopIteration(可捕获处理);
  • 若需更灵活控制(如根据入参返回不同值),改用函数:
mock_get_status.side_effect = lambda player, statuses: (
    "open" if player in ["p1", "p2"] else
    "tosViolation" if player == "p3" else "closed"
)

⚠️ 注意事项与最佳实践

  • 不要依赖 self.model._account_statuses = {...} 工作区绕过 mock:虽能临时规避 API 调用,但丧失了对 _set_thresholds 逻辑(尤其是状态获取流程)的测试覆盖,属于“测试盲区”。
  • 避免混合测试框架:unittest.TestCase 与 @pytest.fixture 共存易引发执行顺序混乱;纯 pytest 更推荐。
  • 验证 mock 行为:除 call_count 外,可用 mock_get_status.call_args_list 检查每次调用的参数,确保逻辑符合预期。
  • 考虑依赖注入(进阶):长期可改造 PlayerAnomalyDetectionModel.__init__,接受 account_status_func 参数,默认为 get_player_account_status,测试时直接传入 mock 函数,彻底解耦。

✅ 总结

问题 正确解法
Mock 不生效 Patch "model.get_player_account_status"(调用方模块路径)
side_effect 无效 确保 patch 作用于 fit() 执行前,且未被其他上下文覆盖
混合框架风险 改用纯 pytest 结构,移除 unittest.TestCase 继承
测试不完整 优先 mock 外部依赖,而非手动设置私有状态

通过精准定位 patch 路径、统一测试风格、善用 side_effect,即可可靠地隔离外部 API,让 fit() 方法在纯本地环境中完成完整逻辑验证。

热门AI工具

更多
WorkBuddy

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

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

Atoms
Atoms Hot

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

DeepSeek

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

豆包大模型

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

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

相关专题

更多
Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

120

2026.09.23

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

60

2026.09.23

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

40

2026.09.23

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

40

2026.09.22

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

40

2026.09.22

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

40

2026.09.22

loomy官网入口地址合集
loomy官网入口地址合集

本专题汇总了 Loomy 桌面 AI 助理的官方入口地址合集及使用指南。提供 macOS 与 Windows 客户端下载 。Loomy 是讯飞推出的桌面级 AI 工作搭子,支持文件整理、数据分析、网页操作及通过飞书/钉钉远程操控电脑,助你高效完成本地办公任务 。

40

2026.09.22

NumPy常见函数使用方法
NumPy常见函数使用方法

本专题整理 NumPy 常见函数使用方法相关教程,覆盖函数大全、参数用法、数组运算、统计聚合、排序处理、where 条件筛选、linspace 创建数列等常用场景,帮助读者快速掌握 NumPy 函数调用思路和实际数据处理技巧。

60

2026.09.22

NumPy性能优化版本更新与常见报错排查
NumPy性能优化版本更新与常见报错排查

本专题整理 NumPy 性能优化、版本更新与常见报错排查相关教程,覆盖向量化计算、广播性能、内存布局、NumPy 2.0 升级、版本兼容冲突、安装导入报错、dtype 溢出、矩阵运算异常和 broadcasting 报错修复,帮助读者系统掌握 NumPy 性能调优与问题定位方法。

80

2026.09.22

热门下载

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

精品课程

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

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