python - 是否有等同于 unittest unittest.TestLoader.loadTestsFromModule() 方法的 Pytest？

Question

我正在开发一个修改测试套件的程序。
在最初的实现中，只支持 unittest 框架。我现在正在尝试添加对 Pytest 的支持。

使用默认的 unitest 模块，我可以将修改后的测试作为 AST 保存在内存中，我将其转换为一个模块，最后，我可以从中创建一个测试套件并运行它。

#create module
code = test.AST.compile()
module = types.ModuleType(test.name)
module.__dict__.update(scope)
exec(code, module.__dict__)
#create suite
suite = unittest.TestLoader().loadTestsFromModule(module)
#run test
suite.run()

根据我在 Pytest 文档中可以找到的所有内容，这似乎是不可能的。
Pytest 需要测试文件才能运行。

有人知道这是否适用于 Pytest 或可能的解决方法？我不想将百分之一的修改测试写入文件系统，因为这比将它们保存在内存中的成本更高。

Answer 1

好的，感谢首席开发人员 nicoddemus我找到了答案，我决定也在这里分享。

其要点如下: 在我的程序内部，我仍然如上所述在内存中创建模块，并将模块压入堆栈。接下来，我使用 pytest.main 调用 pytest:

# Create Module
code = test.AST.compile()
module = types.ModuleType(test.name)
module.__dict__.update(scope)
exec(code, module.__dict__)

modules.append(module)
# Run test
result = pytest.main([test.filename, "-p", "inject_module"])

在 inject_module.py 中，我定义了以下钩子(Hook)，它只是执行默认逻辑，但设置了 Module 的 _obj 属性到我在堆栈上的模块。

def pytest_pycollect_makemodule(path, parent) -> "Module":
    if path.basename == "__init__.py":
        pkg: Package = Package.from_parent(parent, fspath=path)
        return pkg
    mod: Module = Module.from_parent(parent, fspath=path)
    mod._obj = modules.pop()
    return mod