Prefer io over cStringIO and StringIO

2017-04-10 19:43:46 -05:00 · 2017-04-10 19:43:46 -05:00 · fed0d94741
parent 51d5b2d2c9
commit fed0d94741
1 changed files with 3 additions and 3 deletions
--- a/speedtest.py
+++ b/speedtest.py
@ -131,11 +131,11 @@ except ImportError:
    PARSER_TYPE_STR = 'string'

 try:
-    from cStringIO import StringIO
-    BytesIO = None
+    from io import StringIO, BytesIO, TextIOWrapper, FileIO
 except ImportError:
    try:
-        from io import StringIO, BytesIO, TextIOWrapper, FileIO
+        from cStringIO import StringIO
+        BytesIO = None
    except ImportError:
        from StringIO import StringIO
        BytesIO = None