def test_writer(self): writer = csv_py3.DictWriter(self.output, ['a', 'b', 'c']) writer.writeheader() writer.writerow({u'a': u'1', u'b': u'2', u'c': u'☃'}) result = self.output.getvalue() self.assertEqual(result, 'a,b,c\n1,2,☃\n')
def test_writerows(self): writer = csv_py3.DictWriter(self.output, ['a', 'b', 'c'], line_numbers=True) writer.writeheader() writer.writerows([{u'a': u'1', u'b': u'2', u'c': u'☃'}]) result = self.output.getvalue() self.assertEqual(result, 'line_number,a,b,c\n1,1,2,☃\n')