Make Python 2 compatible.

ymoch · Mar 4, 2016 · e2f641f · e2f641f
1 parent 1cfb696
commit e2f641f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/test_dump_as_json.py b/test/test_dump_as_json.py
@@ -3,7 +3,13 @@
 """
 
 import json
-from io import StringIO
+
+# For Python 2 compatibility.
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
+
 from nose.tools import eq_
 from nose.tools import raises