Use io.BytesIO instead of StringsIO.StringsIO

sebastinas · sebastinas · commit 7e0653bd4b29 · 2015-02-04T17:44:46.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/test/test_curtsies_repl.py b/bpython/test/test_curtsies_repl.py
@@ -5,7 +5,6 @@
 from contextlib import contextmanager
 from mock import Mock, patch, MagicMock
 import os
-from StringIO import StringIO
 import sys
 import tempfile
 
@@ -14,6 +13,11 @@
 except ImportError:
     import unittest
 
+try:
+    from io import BytesIO
+except ImportError:
+    from StringIO import StringIO as BytesIO
+
 from bpython.curtsiesfrontend import repl as curtsiesrepl
 from bpython.curtsiesfrontend import interpreter
 from bpython import autocomplete
@@ -201,7 +205,7 @@ def test_list_win_not_visible_and_match_selected_if_one_option(self):
 
 @contextmanager # from http://stackoverflow.com/a/17981937/398212 - thanks @rkennedy
 def captured_output():
-    new_out, new_err = StringIO(), StringIO()
+    new_out, new_err = BytesIO(), BytesIO()
     old_out, old_err = sys.stdout, sys.stderr
     try:
         sys.stdout, sys.stderr = new_out, new_err
