抑制测试功能中的输出

时间:2018-01-16 09:29:41

标签: python python-3.x python-3.6 python-unittest

我想在测试函数中使用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)

0 个答案:

没有答案