def test_counts_numbers(self): table = Table(self.rows, self.columns) new_table = table.counts('one') self.assertEqual(len(new_table.rows), 4) self.assertEqual(len(new_table.columns), 2) self.assertSequenceEqual(new_table.rows[0], [1, 2]) self.assertSequenceEqual(new_table.rows[1], [2, 2]) self.assertSequenceEqual(new_table.rows[2], [None, 1]) self.assertSequenceEqual(new_table.rows[3], [3, 1])
def test_counts_key_func(self): table = Table(self.rows, self.columns) new_table = table.counts(lambda r: r['two']) self.assertEqual(len(new_table.rows), 3) self.assertEqual(len(new_table.columns), 2) self.assertSequenceEqual(new_table.rows[0], ['Y', 1]) self.assertSequenceEqual(new_table.rows[1], ['N', 4]) self.assertSequenceEqual(new_table.rows[2], [None, 1]) self.assertSequenceEqual(new_table.row_names, ['Y', 'N', None])
def test_counts_text(self): table = Table(self.rows, self.columns) new_table = table.counts('two') self.assertEqual(len(new_table.rows), 3) self.assertEqual(len(new_table.columns), 2) self.assertSequenceEqual(new_table.rows[0], ['Y', 1]) self.assertSequenceEqual(new_table.rows[1], ['N', 4]) self.assertSequenceEqual(new_table.rows[2], [None, 1]) self.assertSequenceEqual(new_table.row_names, ['Y', 'N', None])