Adding imports

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_pretty_print.py b/tests/test_pretty_print.py
index 44df156..c2deab8 100755
--- a/tests/test_pretty_print.py
+++ b/tests/test_pretty_print.py
@@ -33,6 +33,13 @@ import hashlib
 import subprocess
 import time
 import platform
+try:
+    from urllib.request import urlopen
+    from urllib.parse import urlencode
+except ImportError:
+    from urllib2 import urlopen
+    from urllib import urlencode
+    
 
 def tryread(filename):
     data = None