def test_large_set(self): """Tests a large set of data""" tabsize = 8 list_data = data.get_party_list() headct = sum([len(v) for v in list_data]) tabct = sum([-(-len(v) // tabsize) for v in list_data]) expected = (headct, tabct) returned = task_01.get_party_stats(list_data, tabsize) self.assertEqual(returned, expected)
def test_table_size(self): """Tests that table size adjusts accordingly""" expected = (14, 6) returned = task_01.get_party_stats(families=self.testdata, table_size=3)
def test_small_set(self): """Tests a small party stats implementation with default table size.""" expected = (14, 4) self.assertEqual(task_01.get_party_stats(self.testdata), expected)