def __init__(self, stdout=True):
    if stdout:
        self._original = sys.stdout
        self._set_stream = self._set_stdout
    else:
        self._original = sys.stderr
        self._set_stream = self._set_stderr
    self._stream = StringIO()
    self._set_stream(self._stream)
def __init__(self, stdout=True):
    if stdout:
        self._original = sys.stdout
        self._set_stream = self._set_stdout
    else:
        self._original = sys.stderr
        self._set_stream = self._set_stderr
    self._stream = StringIO()
    self._set_stream(self._stream)