def test_group_by(self):
table = Table(self.rows, self.column_types, self.column_names)
new_tables = table.group_by('one')
self.assertEqual(len(new_tables), 3)
self.assertIn('a', new_tables.keys())
self.assertIn('b', new_tables.keys())
self.assertIn(None, new_tables.keys())
self.assertSequenceEqual(new_tables['a'].columns['one'], ('a', 'a'))
self.assertSequenceEqual(new_tables['b'].columns['one'], ('b',))
self.assertSequenceEqual(new_tables[None].columns['one'], (None,))
def test_group_by(self):
table = Table(self.rows, self.column_types, self.column_names)
new_tables = table.group_by('one')
self.assertEqual(len(new_tables), 3)
self.assertIn('a', new_tables.keys())
self.assertIn('b', new_tables.keys())
self.assertIn(None, new_tables.keys())
self.assertSequenceEqual(new_tables['a'].columns['one'], ('a', 'a'))
self.assertSequenceEqual(new_tables['b'].columns['one'], ('b', ))
self.assertSequenceEqual(new_tables[None].columns['one'], (None, ))
def test_group_by_bad_column(self):
table = Table(self.rows, self.column_types, self.column_names)
with self.assertRaises(ColumnDoesNotExistError):
table.group_by('bad')
def test_group_by_bad_column(self):
table = Table(self.rows, self.column_types, self.column_names)
with self.assertRaises(ColumnDoesNotExistError):
table.group_by('bad')
