python - 在 Python 中运行单元测试时如何临时隐藏 stdout 或 stderr

Question

我有一个错误的第三方 python 模块，它在导入时输出到 stdout 或 stderr，这破坏了我的单元测试的输出。

如何临时重定向 stdout 以隐藏其输出。

限于 Python 2.5 语法 :)

更新，我忘了提到 sys.stdout 和 sys.__stderr__ 方法在这种情况下不起作用。据我所知，这个有问题的模块正在使用 native 代码。

Answer 1

您还可以使用 mock让您在导入模块时为您修补 sys.stdout 和 sys.stderr。使用此策略的测试模块示例如下:

import os
devnull = open(os.devnull, 'w')

from mock import patch
with patch('sys.stdout', devnull):
    with patch('sys.stderr', devnull):
        import bad_module

# Test cases writen here

其中 bad_module 是在导入时打印到 sys.stdout 和 sys.stderr 的第三方模块。