]> granicus.if.org Git - python/commitdiff
#7960: fix docstrings for captured_output and captured_stdout.
authorEzio Melotti <ezio.melotti@gmail.com>
Sat, 14 May 2011 05:22:47 +0000 (08:22 +0300)
committerEzio Melotti <ezio.melotti@gmail.com>
Sat, 14 May 2011 05:22:47 +0000 (08:22 +0300)
Lib/test/support.py

index baafcc38062417a4b05db60b0bae77049555ff95..b3b1bbdc2d349526990dfe8ee536dbe4814d5eaf 100644 (file)
@@ -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():