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