def test_write_row(self): writer = CSVUnicodeWriter(open(TMPFILE, 'w')) row = [u'Pabl\0231', u'+345667781'] writer.write_row(row) reader = CSVUnicodeReader(open(TMPFILE)) resp = reader.get_rows() self.assertEqual(resp, [row]) os.unlink(TMPFILE)
def test_write_rows(self): writer = CSVUnicodeWriter(open(TMPFILE, 'w')) rows = [[u'Pablo', u'+345667781'], [u'Xabïé', u'+347327211'], [u'Raúls', u'+349494949'], [u'María', u'+312232111'], [u'André', u'+378544522']] writer.write_rows(rows) reader = CSVUnicodeReader(open(TMPFILE)) resp = reader.get_rows() self.assertEqual(resp, rows) os.unlink(TMPFILE)