Jak tymczasowo ukryć stdout lub stderr podczas uruchamiania unittest w Pythonie

Question

Mam wadliwy moduł python trzeciej strony, który jest wyprowadzany do stdout lub stderr podczas importowania i jest to przełamanie wyjścia z moich unittests.

Jak mogę tymczasowo przekierować stdout w celu ukrycia jego danych wyjściowych.

Limit to Python 2.5 składni :)

Aktualizacja, zapomniałem wspomnieć, że sys.stdout i sys.__stderr__ metody nie działają w tym przypadku. O ile mi wiadomo, ten wadliwy moduł używa natywnego kodu.

Answer 1

Można zrobić to tak:

>>> 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

Można również użyć mock pozwolić ci załatać sys.stdout i sys.stderr dla ciebie, gdy moduł jest importowany. Przykładem modułu badawczego, że za pomocą tej strategii będzie:

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 

gdzie bad_module jest moduł osoba trzecia, która drukuje do sys.stdout i sys.stderr gdy jest importowany.