def test_counts_numbers(self): table = Table(self.rows, self.column_names, self.column_types) 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.column_names, self.column_types) 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_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])
def test_counts_key_func(self): table = Table(self.rows, self.column_names, self.column_types) new_table = table.counts(lambda r: r['two']) self.assertIsNot(new_table, table) self.assertColumnNames(new_table, ['group', 'count']) self.assertColumnTypes(new_table, [Text, Number]) self.assertRowNames(new_table, ['Y', 'N', None]) self.assertRows(new_table, [ ['Y', 1], ['N', 4], [None, 1] ])
def test_counts_numbers(self): table = Table(self.rows, self.column_names, self.column_types) new_table = table.counts('one') self.assertIsNot(new_table, table) self.assertColumnNames(new_table, ['one', 'count']) self.assertColumnTypes(new_table, [Number, Number]) self.assertRowNames(new_table, [1, 2, None, 3]) self.assertRows(new_table, [ [1, 2], [2, 2], [None, 1], [3, 1] ])