讲师中心 微信公众号

工具推荐

视频效率加速
立刻MV
立刻MV

低成本歌曲 MV、视频生成

Laper
Laper

剧本策划到成片协作

updream

一键生成高质量推广成片

Seko
Seko

AI短剧一键生成成片

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

老敏姑娘_9537

老敏姑娘_9537

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

|

401人浏览过

|

来源于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 方案(更简洁、无继承冲突):

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载

✅ 推荐修复后的 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工具

更多
WorkBuddy

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

豆包大模型

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

Flowith
Flowith Hot

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

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

码上飞
码上飞 Hot

一款基于自然语言的AI应用开发平台,可根据用户描述自动生成应用程序,帮助个人和团队快速完成应用原型与开发。

超级简历WonderCV

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

DeepSeek

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

相关专题

更多
while的用法
while的用法

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

354

2023.09.25

while的用法
while的用法

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

354

2023.09.25

Go-Zero API开发使用教程
Go-Zero API开发使用教程

本专题聚焦Go-Zero的API开发,拆解.api描述文件的语法结构、路由与分组定义、参数绑定与校验规则,说明统一响应体、JWT鉴权、跨域与接口文档生成的落地方式,并给出网关聚合与接口测试的实践建议。

20

2026.11.12

Go-Zero安装与goctl环境配置教程
Go-Zero安装与goctl环境配置教程

本专题围绕Go-Zero的安装与工具链配置展开,讲清Go版本要求、goctl安装方式、代理镜像、protoc依赖与环境变量设置,说明多版本共存、升级回退与CI流水线安装的做法,并整理安装阶段的常见报错与验证步骤。

20

2026.11.12

Go-Zero零基础入门教程
Go-Zero零基础入门教程

本专题整理Go-Zero的入门知识,从框架定位、环境准备到跑通第一个API服务,讲清目录分层、配置加载、日志与中间件的基本用法,并对比Gin、Kratos的差异,帮助Go开发者建立整体认知,判断它是否适合当前项目。

20

2026.11.12

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

120

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

100

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

100

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

80

2026.11.05

热门下载

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

精品课程

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

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