gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 16:44:56 28 4
gpt4 key购买 nike

我正在开发一个修改测试套件的程序。
在最初的实现中,只支持 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 或可能的解决方法?我不想将百分之一的修改测试写入文件系统,因为这比将它们保存在内存中的成本更高。

最佳答案

好的,感谢首席开发人员 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

关于python - 是否有等同于 unittest unittest.TestLoader.loadTestsFromModule() 方法的 Pytest?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69391625/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com