mirror of
https://github.com/python/cpython.git
synced 2025-08-03 00:23:06 +00:00
Port test_frozen to unittest.
This commit is contained in:
parent
9f2b93e03a
commit
692bbc4790
3 changed files with 49 additions and 24 deletions
|
@ -374,6 +374,22 @@ def transient_internet():
|
|||
return contextlib.nested(time_out, socket_peer_reset, ioerror_peer_reset)
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
def captured_stdout():
|
||||
"""Run the with statement body using a StringIO object as sys.stdout.
|
||||
Example use::
|
||||
|
||||
with captured_stdout() as s:
|
||||
print "hello"
|
||||
assert s.getvalue() == "hello"
|
||||
"""
|
||||
import StringIO
|
||||
orig_stdout = sys.stdout
|
||||
sys.stdout = StringIO.StringIO()
|
||||
yield sys.stdout
|
||||
sys.stdout = orig_stdout
|
||||
|
||||
|
||||
#=======================================================================
|
||||
# Decorator for running a function in a different locale, correctly resetting
|
||||
# it afterwards.
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue