def test_maxfieldsize(self):
# Testing --maxfieldsize for failure. Creating data using str * int.
with open('dummy.csv', 'r') as f:
c = unicsv.UnicodeCSVReader(f, maxfieldsize=9)
with self.assertRaises(unicsv.FieldSizeLimitError):
c.next()
# Now testing higher --maxfieldsize.
with open('dummy.csv', 'r') as f:
c = unicsv.UnicodeCSVReader(f, maxfieldsize=11)
self.assertEqual(['a' * 10], c.next())
def test_maxfieldsize(self):
# Testing --maxfieldsize for failure. Creating data using str * int.
with open('dummy.csv', 'r') as f:
c = unicsv.UnicodeCSVReader(f, maxfieldsize=9)
try:
c.next()
except unicsv.FieldSizeLimitError:
pass
else:
raise AssertionError('Expected unicsv.FieldSizeLimitError')
# Now testing higher --maxfieldsize.
with open('dummy.csv', 'r') as f:
c = unicsv.UnicodeCSVReader(f, maxfieldsize=11)
self.assertEqual(['a' * 10], c.next())
def test_utf16_little(self):
output = StringIO()
writer = unicsv.UnicodeCSVWriter(output, encoding='utf-16-le')
self.assertEqual(writer._eight_bit, False)
writer.writerow(['a', 'b', 'c'])
writer.writerow(['1', '2', '3'])
writer.writerow(['4', '5', u'ʤ'])
written = StringIO(output.getvalue())
reader = unicsv.UnicodeCSVReader(written, encoding='utf-16-le')
self.assertEqual(reader.next(), ['a', 'b', 'c'])
self.assertEqual(reader.next(), ['1', '2', '3'])
self.assertEqual(reader.next(), ['4', '5', u'\u02A4'])
def test_latin1(self):
output = StringIO()
writer = unicsv.UnicodeCSVWriter(output, encoding='latin1')
self.assertEqual(writer._eight_bit, True)
writer.writerow(['a', 'b', 'c'])
writer.writerow(['1', '2', '3'])
writer.writerow(['4', '5', u'©'])
written = StringIO(output.getvalue())
reader = unicsv.UnicodeCSVReader(written, encoding='latin1')
self.assertEqual(reader.next(), ['a', 'b', 'c'])
self.assertEqual(reader.next(), ['1', '2', '3'])
self.assertEqual(reader.next(), ['4', '5', u'©'])
def test_utf16_little(self):
with open('examples/test_utf16_little.csv') as f:
reader = unicsv.UnicodeCSVReader(f, encoding='utf-16')
self.assertEqual(reader.next(), ['a', 'b', 'c'])
self.assertEqual(reader.next(), ['1', '2', '3'])
self.assertEqual(reader.next(), ['4', '5', u'ʤ'])
def test_latin1(self):
with open('examples/test_latin1.csv') as f:
reader = unicsv.UnicodeCSVReader(f, encoding='latin1')
self.assertEqual(reader.next(), ['a', 'b', 'c'])
self.assertEqual(reader.next(), ['1', '2', '3'])
self.assertEqual(reader.next(), ['4', '5', u'©'])
