如何在Python中运行unittest时临时隐藏stdout或stderr

Question

我有一个错误的第三方python模块，它在导入时输出到stdout或stderr，这是打破了我的unittests的输出。

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

限于Python 2.5的语法:)

更新，我忘了提，sys.stdout和sys.__stderr__方法并不在这种情况下工作。据我所知，这个错误的模块正在使用本机代码。

Answer 1

你可以做这样的事情：

>>> import sys, os 
>>> _stderr = sys.stderr 
>>> _stdout = sys.stdout 
>>> null = open(os.devnull,'wb') 
>>> sys.stdout = sys.stderr = null 
>>> print "Bleh" 
>>> sys.stderr = _stderr 
>>> sys.stdout = _stdout 
>>> print "Bleh" 
Bleh 

Answer 2

您还可以使用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被导入时的第三方模块。