diff options
Diffstat (limited to 'numpy/lib/tests/test_utils.py')
| -rw-r--r-- | numpy/lib/tests/test_utils.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/numpy/lib/tests/test_utils.py b/numpy/lib/tests/test_utils.py index 4b386e0b1..8cec1887b 100644 --- a/numpy/lib/tests/test_utils.py +++ b/numpy/lib/tests/test_utils.py @@ -1,10 +1,14 @@ from __future__ import division, absolute_import +import sys from numpy.testing import * import numpy.lib.utils as utils from numpy.lib import deprecate -from StringIO import StringIO +if sys.version_info[0] >= 3: + from io import StringIO +else: + from StringIO import StringIO def test_lookfor(): out = StringIO() |