Python类中正确调用函数的方法

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《Python 类定义时函数调用的正确方法》，聊聊，我们一起来看看吧！

本文详解为何直接 `@patch` 无法影响类属性中提前执行的函数调用，并提供可靠方案：结合 `patch.object` 与 `importlib.reload` 在模块重载前替换目标函数，确保类属性初始化时即使用模拟返回值。

在 Python 单元测试中，若一个函数在类定义阶段（而非实例化时）被调用并赋值给类属性，常规的 @patch 装饰器将失效——因为此时模块已加载、类已构建完成，bar() 的原始返回值 "bar" 已被静态写入 Foo.class_attribute。@patch("foo.bar") 只能拦截后续对 foo.bar 的访问，却无法回溯修改早已计算完毕的类属性。

根本原因在于执行顺序：

1. import foo → 执行 foo.py 模块代码
2. class Foo: 定义开始 → class_attribute = bar() 立即执行（调用真实 bar）
3. 测试运行 @patch("foo.bar") → 此时 Foo 类早已存在，class_attribute 值不可变

✅ 正确解法：在类定义前重置函数行为，并强制重载模块

需分三步操作：

• 使用 patch.object 直接作用于源模块（methods）的目标函数（bar），避免路径歧义；
• 调用 importlib.reload(foo) 重新执行 foo.py 全部代码，触发 class_attribute = bar() 再次执行（此时 bar 已被 mock）；
• 确保 methods 和 foo 模块均在 patch 作用域内被正确引用。

以下是可直接运行的完整测试示例：

# test_foo.py
import importlib
import unittest
from unittest import mock

import foo
import methods  # 必须显式导入，否则 reload 后无法定位原函数


class FooTestCase(unittest.TestCase):
    def test_mock_class_attribute_at_definition_time(self):
        expected = "patched foo"

        # 关键：patch methods.bar（真实定义处），而非 foo.bar
        with mock.patch.object(methods, "bar", return_value=expected):
            # 强制重载 foo 模块，使类重新定义
            importlib.reload(foo)

            # 验证：类属性与实例属性均使用 mock 返回值
            self.assertEqual(foo.Foo.class_attribute, expected)
            self.assertEqual(foo.Foo().class_attribute, expected)

⚠️ 注意事项：

• importlib.reload() 有副作用：会重新执行模块顶层代码，可能影响其他共享状态（如全局变量、单例），建议仅在隔离的测试模块中使用；
• 必须 patch 源模块：@patch("foo.bar") 无效，因 bar 实际来自 methods；正确路径是 mock.patch.object(methods, "bar")；
• 避免循环依赖：确保 methods 在 foo 之前被导入，且测试文件中先 import methods 再 import foo；
• 不适用于 __init_subclass__ 或 __set_name__ 等动态类钩子：本方案仅解决“类定义时函数调用”这一特定场景。

总结：当类属性依赖外部函数调用时，测试的本质不是“绕过”类定义，而是控制定义发生的环境。通过精准 patch 源函数 + 模块重载，我们让类在受控上下文中重建，从而实现真正意义上的行为模拟。

终于介绍完啦！小伙伴们，这篇关于《Python类中正确调用函数的方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布文章相关知识，快来关注吧！