my bad

gawel · Mar 12, 2015 · a90fcc8 · a90fcc8
1 parent 7251205
commit a90fcc8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests.py b/tests.py
@@ -8,7 +8,12 @@
 from unittest import TestCase
 from unittest import skipIf
 from gp.recipe.node import Recipe
-from io import StringIO
+try:
+    # python 2
+    from StringIO import StringIO
+except:
+    # python 3
+    from io import StringIO
 
 PY3 = bool(sys.version_info[0] == 3)