python - 创建一个 python 单元测试，其中缺少模块是伪造的

Question

我有一段在许多不同系统上运行的代码。在某些系统上，模块 pyodbc不存在所以我退回到 pymssql .

我想在安装了这两个模块的环境中测试回退过程。

增加难度:pyodbc未在我的代码中的任何地方导入，仅在 sqlalchemy 内.

图书馆的细节无关紧要。我只想避免必须运行 pip 来删除和替换 pyodbc模块，然后必须测试两次。

有没有办法欺骗丢失的模块？

Answer 1

您可以通过使用 sys.modules 来假装没有模块。 :

>>> import sys
>>> sys.modules['pyodbc'] = None
>>> import pyodbc
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: import of 'pyodbc' halted; None in sys.modules

(盗自 this answer)