From 60a9c69c5e7b4ed4aef3903a7fa031d08c6b1979 Mon Sep 17 00:00:00 2001 From: Benjamin Peterson Date: Tue, 29 Dec 2009 00:37:04 +0000 Subject: fix test on py3 --- Lib/lib2to3/tests/test_main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Lib/lib2to3/tests/test_main.py b/Lib/lib2to3/tests/test_main.py index 23d1b56..2485f7a 100644 --- a/Lib/lib2to3/tests/test_main.py +++ b/Lib/lib2to3/tests/test_main.py @@ -25,12 +25,12 @@ class TestMain(unittest.TestCase): def test_unencodable_diff(self): input_stream = io.StringIO("print 'nothing'\nprint u'über'\n") - out = io.StringIO() + out = io.BytesIO() out_enc = codecs.getwriter("ascii")(out) err = io.StringIO() ret = self.run_2to3_capture(["-"], input_stream, out_enc, err) self.assertEqual(ret, 0) - output = out.getvalue() + output = out.getvalue().decode("ascii") self.assertTrue("-print 'nothing'" in output) self.assertTrue("WARNING: couldn't encode 's diff for " "your terminal" in err.getvalue()) -- cgit v0.12