我想在测试函数中使用print
来抑制输出,同时使用测试函数上的装饰器以安全的方式保持测试函数的输出。基本上:
script.py
def func():
print("script")
此模块在:
中调用test_script.py
import script
@some_decorator
def test_func():
print("test_script")
script.func()
test_func()
some_decorator
应该是什么,test_script
在没有script
的情况下打印?我可以根据需要在任一脚本上导入任何库,最好是从标准库中导入。
谢谢!
编辑:
我使用importlib
导入script.py
,因此导入行应该是这样的:
import importlib
file_name = "script"
script = importlib.import_module(file_name)