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