Be more careful with the shuffling of stdout.
diff --git a/Lib/test/autotest.py b/Lib/test/autotest.py
index 4b15c26..aadd170 100644
--- a/Lib/test/autotest.py
+++ b/Lib/test/autotest.py
@@ -89,12 +89,13 @@
def do_one_test(t, outdir):
filename = os.path.join(outdir, t)
real_stdout = sys.stdout
+ if generate:
+ print 'Generating:', filename
+ fake_stdout = open(filename, 'w')
+ else:
+ fake_stdout = Compare(filename)
try:
- if generate:
- print 'Generating:', filename
- sys.stdout = open(filename, 'w')
- else:
- sys.stdout = Compare(filename)
+ sys.stdout = fake_stdout
print t
unload(t)
try:
@@ -104,8 +105,8 @@
sys.stderr.write(msg+': Un-installed'
' optional module?\n')
finally:
- sys.stdout.close()
sys.stdout = real_stdout
+ fake_stdout.close()