diff options
author | Ezio Melotti <ezio.melotti@gmail.com> | 2011-05-14 05:22:47 (GMT) |
---|---|---|
committer | Ezio Melotti <ezio.melotti@gmail.com> | 2011-05-14 05:22:47 (GMT) |
commit | fc778fd06753178a817b4aa51ddb8dd503fe7ea8 (patch) | |
tree | 7755203e6caba866ba84941bd702e9b45463ebda | |
parent | 3c0d8a1cc728a92671a9f3d4cbf6636453707385 (diff) | |
download | cpython-fc778fd06753178a817b4aa51ddb8dd503fe7ea8.zip cpython-fc778fd06753178a817b4aa51ddb8dd503fe7ea8.tar.gz cpython-fc778fd06753178a817b4aa51ddb8dd503fe7ea8.tar.bz2 |
#7960: fix docstrings for captured_output and captured_stdout.
-rw-r--r-- | Lib/test/support.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/Lib/test/support.py b/Lib/test/support.py index baafcc3..b3b1bbd 100644 --- a/Lib/test/support.py +++ b/Lib/test/support.py @@ -712,14 +712,8 @@ def transient_internet(resource_name, *, timeout=30.0, errnos=()): @contextlib.contextmanager def captured_output(stream_name): - """Run the 'with' statement body using a StringIO object in place of a - specific attribute on the sys module. - Example use (with 'stream_name=stdout'):: - - with captured_stdout() as s: - print("hello") - assert s.getvalue() == "hello" - """ + """Return a context manager used by captured_stdout and captured_stdin + that temporarily replaces the sys stream *stream_name* with a StringIO.""" import io orig_stdout = getattr(sys, stream_name) setattr(sys, stream_name, io.StringIO()) @@ -729,6 +723,12 @@ def captured_output(stream_name): setattr(sys, stream_name, orig_stdout) def captured_stdout(): + """Capture the output of sys.stdout: + + with captured_stdout() as s: + print("hello") + self.assertEqual(s.getvalue(), "hello") + """ return captured_output("stdout") def gc_collect(): |