1 files changed, 8 insertions, 5 deletions
diff --git a/numpy/lib/tests/test__datasource.py b/numpy/lib/tests/test__datasource.py
index e784e4296..ccd7405fe 100644
--- a/numpy/lib/tests/test__datasource.py
+++ b/numpy/lib/tests/test__datasource.py
@@ -1,17 +1,20 @@
 from __future__ import division, absolute_import
 
 import os
+import urllib2
+import sys
+import numpy.lib._datasource as datasource
 from tempfile import mkdtemp, mkstemp, NamedTemporaryFile
 from shutil import rmtree
-from urlparse import urlparse
 from urllib2 import URLError
-import urllib2
-
+from numpy.compat import asbytes
 from numpy.testing import *
 
-from numpy.compat import asbytes
 
-import numpy.lib._datasource as datasource
+if sys.version_info[0] >= 3:
+    from urllib.parse import urlparse
+else:
+    from urlparse import urlparse
 
 def urlopen_stub(url, data=None):
     '''Stub to replace urlopen for testing.'''