python - 如何测试 Python 中缺少特定库的情况

Question

我有一些软件包对其他软件包具有软依赖性，并回退到默认(简单)实现。

问题是使用单元测试很难测试。我可以设置单独的虚拟环境，但这很难管理。

是否有一个包或一种方法可以实现以下目标:有

import X

照常工作，但是

hide_package('X')
import X

将引发 ImportError。

我的代码的回退部分一直有错误，因为很难对其进行测试。

Answer 1

它看起来有点脏，但你可以重写 __import__ 内建函数:

save_import = __builtin__.__import__
def my_import(name, *rest):
    if name=="hidden":
        raise ImportError, "Hidden package"
    return save_import(name, *rest)
__builtin__.__import__ = my_import

顺便说一句，你读过 PEP 302 了吗？看来您可以使用 import hooks 建立更健壮的机制。