Merge pull request package-url#8 from haikoschol/py3-tests-fix

pombredanne · web-flow · commit fcf24e26819a · 2018-02-07T10:56:51.000+01:00
Only convert test function names to byte strings on Python 2
diff --git a/test_purl.py b/test_purl.py
@@ -37,10 +37,12 @@
     unicode
     str = unicode
     basestring = basestring
+    py2 = True
 except NameError:
     # Python 3
     unicode = str
     basestring = (bytes, str,)
+    py2 = False
 
 
 def create_test_function(description, purl, canonical_purl, is_invalid,
@@ -113,7 +115,7 @@ def python_safe_name(s):
     s = s.lower()
     s = no_punctuation(' ', s)
     s = '_'.join(s.split())
-    if isinstance(s, unicode):
+    if py2 and isinstance(s, unicode):
         s = s.encode('ascii', 'ignore')
     return s
 
