def test_select_no_columns_is_error(self): input_csv = u'''\ a,b a,1 a,2 x,3''' with self.assertRaises(Exception): m.select(records_from_text(input_csv), [])
def assert_selects(self, result_csv, input_csv, columns): self.assertListEqual( records_from_text(result_csv), list( m.select( records_from_text(input_csv), columns.split(u','))))
def test_select_one_column(self): input_csv = u'''\ a,b,c a,1,c a,2,d x,3,x''' self.assertListEqual( [u'1', u'2', u'3'], list( m.select( records_from_text(input_csv), [u'b'])))