Làm thế nào để tạm thời ẩn stdout hoặc stderr trong khi chạy một unittest trong Python

Question

Tôi có một mô-đun python bên thứ ba bị lỗi được đầu ra để stdout hoặc stderr trong khi nó được nhập khẩu và điều này là phá vỡ đầu ra của unittests của tôi.

Làm cách nào để tạm thời chuyển hướng stdout để ẩn đầu ra của nó.

Giới hạn tới Python 2.5 cú pháp :)

Cập nhật, tôi quên đề cập đến rằng sys.stdout và sys.__stderr__ phương pháp này không làm việc trong trường hợp này. Theo như tôi biết mô-đun bị lỗi này đang sử dụng mã gốc.

Answer 1

Bạn có thể làm điều đó một cái gì đó như thế này:

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

Bạn cũng có thể sử dụng mock để cho bạn vá sys.stdout và sys.stderr cho bạn khi module được nhập khẩu. Một ví dụ về một mô-đun thử nghiệm rằng việc sử dụng chiến lược này sẽ là:

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 

nơi bad_module là module bên thứ ba đang in để sys.stdout và sys.stderr khi đang được nhập khẩu.