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'])))
