def test_dump(self):
matrix = [
['あ', 'い', 'う'],
['か', 'き'],
['"']
]
expected = '"あ","い","う"\n"か","き"\n""""'.encode('utf8', 'ignore')
self.assertEqual(CSVDumper.dump(matrix, 'utf8'), expected)
def dump_as_csv(self, encoding='sjis'):
rows = []
for question in self.question_set.all():
rows.append([question.content])
if question.is_selection_form_type():
rows.append([None, 'Answer', 'Percentage', 'Number'])
for selectiveanswer in question.selectiveanswer_set.all():
rows.append([
None,
selectiveanswer.content,
selectiveanswer.percentage(),
selectiveanswer.num
])
else:
rows.append([None, 'Answer'])
for textanswer in question.textanswer_set.all():
rows.append([None, textanswer.content])
return CSVDumper.dump(rows, encoding)
