diff options
Diffstat (limited to 'numpy/lib/tests/test_format.py')
| -rw-r--r-- | numpy/lib/tests/test_format.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/numpy/lib/tests/test_format.py b/numpy/lib/tests/test_format.py index ec035ca28..f6a6c87ce 100644 --- a/numpy/lib/tests/test_format.py +++ b/numpy/lib/tests/test_format.py @@ -2,7 +2,11 @@ r''' Test the .npy file format. Set up: - >>> from cStringIO import StringIO + >>> import sys + >>> if sys.version_info[0] >= 3: + ... from io import BytesIO as StringIO + ... else: + ... from cStringIO import StringIO >>> from numpy.lib import format >>> >>> scalars = [ @@ -275,11 +279,15 @@ Test the header writing. import sys -from cStringIO import StringIO import os import shutil import tempfile +if sys.version_info[0] >= 3: + from io import BytesIO as StringIO +else: + from cStringIO import StringIO + import numpy as np from numpy.testing import * |