]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
#7960: fix docstrings for captured_output and captured_stdout.
authorEzio Melotti <ezio.melotti@gmail.com>
Sat, 14 May 2011 05:19:43 +0000 (08:19 +0300)
committerEzio Melotti <ezio.melotti@gmail.com>
Sat, 14 May 2011 05:19:43 +0000 (08:19 +0300)
Lib/test/test_support.py

index e2377093664d2b42c9601085c2742aaa585eba75..81bb3ca017d46e060873f72f9d525e484d924196 100644 (file)
@@ -814,14 +814,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 StringIO
     orig_stdout = getattr(sys, stream_name)
     setattr(sys, stream_name, StringIO.StringIO())
@@ -831,6 +825,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 captured_stdin():