Fix python 2 compatibility issue

lenarother · Jan 20, 2019 · 0e5d1c1 · 0e5d1c1
1 parent 06966bc
commit 0e5d1c1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/client/test_commands.py b/tests/client/test_commands.py
@@ -1,4 +1,4 @@
-from io import StringIO
+import sys
 
 import mock
 import pytest
@@ -7,6 +7,12 @@
 from requests.exceptions import ConnectionError
 
 
+if sys.version_info[0] < 3:
+    from StringIO import StringIO
+else:
+    from io import StringIO
+
+
 @pytest.mark.django_db
 class TestSyncPerms: