贝利信息

如何在 pytest 中绕过 joblib 缓存直接测试原始函数

日期:2026-01-24 00:00 / 作者:聖光之護

在 pytest 单元测试中,可通过访问 `joblib` 缓存函数的 `.func` 属性跳过缓存,直接调用原始逻辑,确保每次测试都执行真实计算而非读取缓存结果。

当使用 joblib.Memory 装饰函数(如 @memory.cache)时,装饰器返回的是一个 MemorizedFunc 实例,而非原始函数本身。该实例封装了缓存逻辑,但同时保留了对原始函数的引用——即 .func 属性。这是 joblib 官方支持的访问方式,无需 mock、patch 或修改全局配置,简洁、安全且符合设计意图。

以下是一个完整示例:

from joblib import Memory

# 生产环境使用的缓存配置
memory = Memory(location="cache")

@memory.cache
def func(a):
    print("Executing real computation...")  # 可用于验证是否

被跳过 return a ** 2 def test_func_bypasses_cache(): # 直接调用原始函数,完全绕过缓存机制 result = func.func(2) assert result == 4 # 注意:此处不会输出 "Executing real computation..." 两次, # 因为 .func 不触发缓存逻辑,也不写入/读取磁盘

优势说明

⚠️ 注意事项

总之,func.func 是专为测试而设的轻量级“逃生通道”,是绕过 joblib 缓存最直接、最推荐的实践方式。