diff options
| -rw-r--r-- | Lib/lib2to3/tests/test_main.py | 4 |
1 files 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 <stdin>'s diff for " "your terminal" in err.getvalue()) |