Upgrade python 3 decode error

This commit is contained in:
asadazam93
2019-09-27 16:07:36 +05:00
parent 27aa7e4abf
commit 7a43721f81
2 changed files with 6 additions and 2 deletions

View File

@@ -5,6 +5,7 @@ recommender system
from __future__ import absolute_import
import codecs
import itertools
import json
from six import StringIO
@@ -650,7 +651,10 @@ class TestRecommenderFileUploading(TestRecommender):
happens or is rejected as expected.
"""
if 'magic_number' in test_case:
f_handler = StringIO(test_case['magic_number'].decode('hex'))
if six.PY2:
f_handler = StringIO(codecs.decode(test_case['magic_number'], 'hex_codec'))
else:
f_handler = StringIO(codecs.decode(test_case['magic_number'], 'hex_codec').decode('utf-8'))
elif content is not None:
f_handler = StringIO(json.dumps(content, sort_keys=True))
else: