Use the new io.StringIO in python3

a5df282c · Luke Macken · ee9e197f · a5df282c
Commit a5df282c authored Jun 21, 2013 by Luke Macken
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

pyrasite/tests/test_cli.py pyrasite/tests/test_cli.py +4 -1

No files found.
--- a/pyrasite/tests/test_cli.py
+++ b/pyrasite/tests/test_cli.py
@@ -18,7 +18,10 @@
 import os
 import sys
-from StringIO import StringIO
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
 from pyrasite.main import main