use six.string_types instead of str to handle unicode in Python2

This commit is contained in:
Nate Prewitt
2017-03-14 09:45:56 -06:00
parent e2d7605590
commit d124103ab0
2 changed files with 7 additions and 1 deletions
+2 -1
View File
@@ -5,6 +5,7 @@ import tempfile
import parse
import requests
import requirements
import six
try:
from HTMLParser import HTMLParser
@@ -81,7 +82,7 @@ def convert_deps_to_pip(deps, r=True):
for dep in deps.keys():
# Default (e.g. '>1.10').
extra = deps[dep] if isinstance(deps[dep], str) else ''
extra = deps[dep] if isinstance(deps[dep], six.string_types) else ''
version = ''
# Get rid of '*'.
+5
View File
@@ -56,6 +56,11 @@ class TestUtils:
deps = pipenv.utils.convert_deps_to_pip(deps, r=False)
assert deps[0] == 'FooProject==1.2 --hash=sha256:2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824'
# test unicode values
deps = {u'django': u'==1.10'}
deps = pipenv.utils.convert_deps_to_pip(deps, r=False)
assert deps[0] == 'django>1.10'
def test_convert_from_pip(self):