在 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 缓存最直
