def test_table_from_csv(self): import csvkit from agate import table table.csv = csvkit if six.PY2: table = Table.from_csv('examples/test.csv', self.tester, encoding='utf8') else: table = Table.from_csv('examples/test.csv', self.tester) self.assertSequenceEqual(table.get_column_names(), ['one', 'two', 'three']) self.assertSequenceEqual(tuple(map(type, table.get_column_types())), [Number, Number, Text]) self.assertEqual(len(table.columns), 3) self.assertSequenceEqual(table.rows[0], [1, 4, 'a']) self.assertSequenceEqual(table.rows[1], [2, 3, 'b']) self.assertSequenceEqual(table.rows[2], [None, 2, u'👍'])
def test_get_column_types(self): table = Table(self.rows, self.columns) self.assertSequenceEqual(table.get_column_types(), [t for n, t in self.columns])