py3 Assetion Error
This commit is contained in:
Ayub khan
2019-08-23 15:31:29 +05:00
parent 81a99e4d32
commit 5b984ffb5c
2 changed files with 3 additions and 3 deletions

View File

@@ -32,11 +32,11 @@ def create_csv_response(filename, header, datarows):
quotechar='"',
quoting=csv.QUOTE_ALL)
encoded_header = [six.text_type(s).encode('utf-8') for s in header]
encoded_header = [six.text_type(s) for s in header]
csvwriter.writerow(encoded_header)
for datarow in datarows:
encoded_row = [six.text_type(s).encode('utf-8') for s in datarow]
encoded_row = [six.text_type(s) for s in datarow]
csvwriter.writerow(encoded_row)
return response

View File

@@ -19,7 +19,7 @@ class TestAnalyticsCSVS(TestCase):
res = create_csv_response('robot.csv', header, datarows)
self.assertEqual(res['Content-Type'], 'text/csv')
self.assertEqual(res['Content-Disposition'], u'attachment; filename={0}'.format('robot.csv'))
self.assertEqual(res.content.strip(), '"Name","Email"')
self.assertEqual(res.content.strip().decode('utf-8'), '"Name","Email"')
def test_create_csv_response(self):
header = ['Name', 'Email']