def test_limit_columns(self): "Column definitions are passed to rows" t = TableFu(self.csv_file) t.columns = ['Author', 'Style'] self.assertEqual( str(t[0]), 'Samuel Beckett, Modernism' )
def test_set_columns(self): "Set new columns for a table" t = TableFu(self.csv_file) columns = ['Style', 'Author'] t.columns = columns self.assertEqual(t.columns, columns)