From f65b3a1495fb9f07cbf83e6eaae67074a1e32df3 Mon Sep 17 00:00:00 2001 From: Thomas Kluyver Date: Thu, 24 Oct 2013 11:04:49 -0700 Subject: [PATCH] Use StringIO.StringIO on Python 2. io.StringIO is strict about unicode, StringIO.StringIO isn't. --- IPython/utils/tests/test_io.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/IPython/utils/tests/test_io.py b/IPython/utils/tests/test_io.py index c3a099ce6..4818c0740 100644 --- a/IPython/utils/tests/test_io.py +++ b/IPython/utils/tests/test_io.py @@ -15,14 +15,18 @@ from __future__ import print_function import sys -from io import StringIO from subprocess import Popen, PIPE import unittest import nose.tools as nt from IPython.utils.io import Tee, capture_output -from IPython.utils.py3compat import doctest_refactor_print +from IPython.utils.py3compat import doctest_refactor_print, PY3 + +if PY3: + from io import StringIO +else: + from StringIO import StringIO #----------------------------------------------------------------------------- # Tests