Py3K

apibyexample · Oct 12, 2016 · c6e2790 · c6e2790
1 parent d482121
commit c6e2790
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/abe/unittest.py b/abe/unittest.py
@@ -1,5 +1,8 @@
 import os
-import urlparse
+try:
+    from urlparse import parse_qs
+except ImportError:
+    from urllib.parse import parse_qs
 
 from .mocks import AbeMock
 from .utils import normalize, subkeys
@@ -127,7 +130,7 @@ def assert_headers_contain(self, response_data, spec_data):
             )
 
     def assert_query_params_equal(self, request_data, spec_data):
-        qs = urlparse.parse_qs(request_data)
+        qs = parse_qs(request_data)
         for k, expected_value in spec_data.items():
             try:
                 actual_value = qs[k]