0

0

如何在单元测试中正确模拟类实例内部调用的函数

花韻仙語

花韻仙語

发布时间:2026-01-06 16:02:02

|

381人浏览过

|

来源于php中文网

原创

如何在单元测试中正确模拟类实例内部调用的函数

本文详解如何在 python 单元测试中精准 mock 被类方法内部调用的外部函数(如 `get_player_account_status`),重点解决路径指定错误、作用域不匹配及 `side_effect` 动态返回等常见问题,并提供可复现的修复方案。

在测试 PlayerAnomalyDetectionModel.fit() 时,你遇到的核心问题并非 side_effect 使用不当,而是 mock 路径错误——这是 unittest.mock.patch 最常见的陷阱。关键原则是:必须 patch 函数被“导入并使用”的位置,而非其“定义位置”

回顾你的代码结构:

  • get_player_account_status 在 get_player_labels.py 中定义;
  • 但在 model.py 中通过 from get_player_labels import get_player_account_status 导入并直接调用;
  • 因此,PlayerAnomalyDetectionModel._set_thresholds() 内部实际引用的是 model.get_player_account_status(即导入后的符号),而非 get_player_labels.get_player_account_status。

✅ 正确 patch 路径应为:'model.get_player_account_status'
❌ 错误路径(导致 mock 失效):'get_player_labels.get_player_account_status' 或 'model.get_player_labels.get_player_account_status'

此外,你的测试类混合了 pytest fixture 和 unittest.TestCase,存在兼容性风险(@mock.patch 装饰器对 unittest.TestCase 子类有效,但 @pytest.fixture(autouse=True) 在 unittest 类中行为不可靠)。建议统一风格,推荐纯 pytest 方案(更简洁、无继承冲突):

Anyword
Anyword

AI文案写作助手和文本生成器,具有可预测结果的文案 AI

下载

✅ 推荐修复后的 test_model.py 片段:

import pandas as pd
import pytest
from unittest.mock import patch
from model import PlayerAnomalyDetectionModel


@pytest.fixture
def sample_train_data():
    # 构造示例数据(此处省略具体实现)
    return pd.DataFrame({"player": ["p1", "p2", "p3"], "score": [1.0, 2.0, 3.0]})


# 使用 pytest 风格:直接 patch model 模块中的函数
def test_fit_with_mocked_account_status(sample_train_data):
    model = PlayerAnomalyDetectionModel()

    # 指定正确的 patch 路径:model 模块内导入的函数名
    with patch('model.get_player_account_status') as mock_get_status:
        # 配置 side_effect 实现每次调用返回不同值
        mock_get_status.side_effect = [
            'open', 'open', 'tosViolation', 'tosViolation', 'tosViolation', 'closed',
            'open', 'open', 'tosViolation', 'tosViolation', 'tosViolation', 'closed'
        ]

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

        # 断言 mock 被调用(可选验证)
        assert mock_get_status.call_count == 12  # 根据实际逻辑调整预期次数

⚠️ 关键注意事项:

  • 路径必须精确:若 model.py 中写的是 from get_player_labels import get_player_account_status,则 patch 'model.get_player_account_status';若写的是 import get_player_labels 且调用 get_player_labels.get_player_account_status(...),才需 patch 'model.get_player_labels.get_player_account_status'。
  • 避免混合测试框架:unittest.TestCase 与 @pytest.fixture 共存易引发 setup/teardown 行为异常。纯 pytest 更可靠。
  • side_effect 用法正确:你原写法本身无误——列表形式 side_effect = [...] 会按顺序逐次返回,完全满足“每次调用不同值”的需求。
  • Workaround 的局限性:手动预设 _account_statuses 虽能绕过 API 调用,但会使测试脱离真实执行路径(例如跳过了 _set_thresholds 中的 while 循环逻辑),降低测试保真度。应优先采用正确 mock。

✅ 总结

正确 mock 的三要素:

  1. 定位准:patch 函数在被测代码所在模块中的导入名
  2. 作用域对:patch 必须在被测方法执行前生效(with patch 或装饰器);
  3. 验证全:通过 call_count、assert_called_with() 等校验 mock 行为是否符合预期。

遵循以上原则,即可稳定拦截类内部的外部函数调用,彻底避免真实 API 请求,构建健壮、快速、可重复的单元测试。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
while的用法
while的用法

while的用法是“while 条件: 代码块”,条件是一个表达式,当条件为真时,执行代码块,然后再次判断条件是否为真,如果为真则继续执行代码块,直到条件为假为止。本专题为大家提供while相关的文章、下载、课程内容,供大家免费下载体验。

105

2023.09.25

while的用法
while的用法

while的用法是“while 条件: 代码块”,条件是一个表达式,当条件为真时,执行代码块,然后再次判断条件是否为真,如果为真则继续执行代码块,直到条件为假为止。本专题为大家提供while相关的文章、下载、课程内容,供大家免费下载体验。

105

2023.09.25

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

26

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

68

2026.03.05

PHP高性能API设计与Laravel服务架构实践
PHP高性能API设计与Laravel服务架构实践

本专题围绕 PHP 在现代 Web 后端开发中的高性能实践展开,重点讲解基于 Laravel 框架构建可扩展 API 服务的核心方法。内容涵盖路由与中间件机制、服务容器与依赖注入、接口版本管理、缓存策略设计以及队列异步处理方案。同时结合高并发场景,深入分析性能瓶颈定位与优化思路,帮助开发者构建稳定、高效、易维护的 PHP 后端服务体系。

164

2026.03.04

AI安装教程大全
AI安装教程大全

2026最全AI工具安装教程专题:包含各版本AI绘图、AI视频、智能办公软件的本地化部署手册。全篇零基础友好,附带最新模型下载地址、一键安装脚本及常见报错修复方案。每日更新,收藏这一篇就够了,让AI安装不再报错!

84

2026.03.04

Swift iOS架构设计与MVVM模式实战
Swift iOS架构设计与MVVM模式实战

本专题聚焦 Swift 在 iOS 应用架构设计中的实践,系统讲解 MVVM 模式的核心思想、数据绑定机制、模块拆分策略以及组件化开发方法。内容涵盖网络层封装、状态管理、依赖注入与性能优化技巧。通过完整项目案例,帮助开发者构建结构清晰、可维护性强的 iOS 应用架构体系。

113

2026.03.03

C++高性能网络编程与Reactor模型实践
C++高性能网络编程与Reactor模型实践

本专题围绕 C++ 在高性能网络服务开发中的应用展开,深入讲解 Socket 编程、多路复用机制、Reactor 模型设计原理以及线程池协作策略。内容涵盖 epoll 实现机制、内存管理优化、连接管理策略与高并发场景下的性能调优方法。通过构建高并发网络服务器实战案例,帮助开发者掌握 C++ 在底层系统与网络通信领域的核心技术。

29

2026.03.03

Golang 测试体系与代码质量保障:工程级可靠性建设
Golang 测试体系与代码质量保障:工程级可靠性建设

Go语言测试体系与代码质量保障聚焦于构建工程级可靠性系统。本专题深入解析Go的测试工具链(如go test)、单元测试、集成测试及端到端测试实践,结合代码覆盖率分析、静态代码扫描(如go vet)和动态分析工具,建立全链路质量监控机制。通过自动化测试框架、持续集成(CI)流水线配置及代码审查规范,实现测试用例管理、缺陷追踪与质量门禁控制,确保代码健壮性与可维护性,为高可靠性工程系统提供质量保障。

79

2026.02.28

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.8万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.8万人学习

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

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