有没有一种方法可以对使用的导入进行单元测试？

Question

有没有一种方法可以对 Python 文件中导入的模块进行单元测试（有点像 Java 中的 ArchUnit）？背景是实现六边形架构并希望确保域模型不会导入驻留在适配器中的任何代码。如果存在禁止的导入，我希望单元测试失败。

例如，我可能想测试 foo.bar.domain 中没有模块从 foo.bar.adapter 导入任何内容。应允许从 foo.bar.adapter 内导入 foo.bar.domain。

这在 Python 中可行吗？实现这一目标的最佳方法是什么？

Answer 1

您可以使用 -Ximporttime Python 标志来跟踪导入。我不完全确定在您的情况下查找禁止导入的逻辑是什么，但这里有一个可能有帮助的愚蠢示例脚本：

import subprocess
import sys

process = subprocess.run(
    ('python3', '-Ximporttime', '-c', 'import ' + 'shlex'),
        stdout=subprocess.DEVNULL,
        stderr=subprocess.PIPE,
        encoding='utf-8',
    )

blacklisted_imports = {'enum', 're', 'zipfile'}

data = [
    x.rpartition('|')[2].strip() for x in process.stderr.split('\n')
]
for import_ in data:
    if import_ in blacklisted_imports:
        print('found bad import:', import_)

输出：

found bad import: enum
found bad import: re

Answer 2

我不知道针对这种特定情况是否存在测试方法，但有人可能知道更多。我想到的一件事是 try-catch ，检查其他模块中的方法是否可以调用方法。另一种黑客方法是在每个模块的全局上下文中添加自定义字符串常量，如果它们存在，您就知道子模块导入/使用了其他模块。

详细了解堆栈溢出帖子。