tests: fix capture_stdio() for Python 2.7

geertj · Jun 14, 2017 · 1d77ca4 · 1d77ca4
1 parent e56c630
commit 1d77ca4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/support.py b/tests/support.py
@@ -157,9 +157,9 @@ class capture_stdio(object):
 
     def __enter__(self):
         self.saved_stdout = sys.stdout
-        sys.stdout = io.StringIO()
+        sys.stdout = io.StringIO() if six.PY3 else io.BytesIO()
         self.saved_stderr = sys.stderr
-        sys.stderr = io.StringIO()
+        sys.stderr = io.StringIO() if six.PY3 else io.BytesIO()
         return (sys.stdout, sys.stderr)
 
     def __exit__(self, typ, val, tb):