def test_row_iterability_and_string_representation(self): text = "1,2,3\n4,5,6" reader = FancyReader(text.splitlines(), fieldnames=['a', 'b', 'c']) rows = list(reader) self.assertEqual(tuple(rows[0]), ('1', '2', '3')) self.assertEqual(tuple(rows[1]), ('4', '5', '6')) self.assertEqual(repr(rows[0]), "Row(a='1', b='2', c='3')")
def test_line_num(self): text = "a,b,c\n1,2,3\n4,5,6" # Reader with fieldnames reader = FancyReader( text.splitlines(), fieldnames=['a', 'b', 'c'], ) self.assertEqual(reader.line_num, 0) next(reader) self.assertEqual(reader.line_num, 1) next(reader) self.assertEqual(reader.line_num, 2) # Reader without fieldnames reader = FancyReader(text.splitlines()) next(reader) self.assertEqual(reader.line_num, 2) next(reader) self.assertEqual(reader.line_num, 3)
def test_accessing_attributes_from_rows(self): text = "1,2,3\n4,5,6" reader = FancyReader(text.splitlines(), fieldnames=['a', 'b', 'c']) row1, row2 = reader self.assertEqual(row1.a, '1') self.assertEqual(row1.b, '2') self.assertEqual(row1.c, '3') self.assertEqual(row2.a, '4') self.assertEqual(row2.b, '5') self.assertEqual(row2.c, '6')
def test_no_fieldnames_specified(self): # Test fieldnames argument my_file = StringIO("a,b,c\r\n1,2,3\r\n4,5,6\r\n") reader = FancyReader(my_file) rows = list(reader) self.assertEqual(rows[0].a, '1') self.assertEqual(rows[0].b, '2') self.assertEqual(rows[0].c, '3') self.assertEqual(rows[1].a, '4') self.assertEqual(rows[1].b, '5') self.assertEqual(rows[1].c, '6')
def test_many_columns_and_rows(self): text = dedent(""" Julia,Spender,purple,Two in the hand is worth one in the fridge Sarah,Taylor,green,"Learn from yesterday, live for today" Gary,Richter,blue,Be someone you would be proud to know Kathleen,Blocker,red,Live everyday like it's your last Angelo,Griffith,pink,Don't do today what you could do tomorrow """).lstrip() reader = FancyReader( text.splitlines(), fieldnames=['first', 'last', 'color', 'saying'], ) rows = list(reader) self.assertEqual(len(rows), 5) self.assertEqual(rows[4].color, "pink") self.assertEqual( rows[1].saying, "Learn from yesterday, live for today", ) self.assertEqual(rows[2].first, "Gary") self.assertEqual(rows[3].last, "Blocker")
def test_with_files(self): reader = FancyReader( StringIO("1,2,3\r\n4,5,6\r\n"), fieldnames=['a', 'b', 'c'], ) self.assertEqual(len(list(reader)), 2)
def test_lazy_looping(self): my_file = StringIO("1,2,3\r\n4,5,6\r\n") reader = FancyReader((x for x in my_file), fieldnames=['a', 'b', 'c']) self.assertEqual(my_file.tell(), 0) next(reader) self.assertEqual(list(my_file), ["4,5,6\r\n"])
def test_with_list(self): text = "1,2,3\n4,5,6" reader = FancyReader(text.splitlines(), fieldnames=['a', 'b', 'c']) self.assertEqual(len(list(reader)), 2)