3K: lib: fix some bytes/str issues in _format.py and its tests

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 *