class TestCountCases(unittest.TestCase):
def setUp(self):
self.problems = [Mock(cases=[i for i in range(1)], problem_id=('17', '1')),
Mock(cases=[i for i in range(3)], problem_id=('17', '2')),
Mock(cases=[i for i in range(4)], problem_id=('17', '3'))]
self.counter = CasesCounter(self.problems)
def test_get_stat_data(self):
self.assertEqual(self.counter.get_stat_data(), {('17', '1'): 1,
('17', '2'): 3,
('17', '3'): 4
}
)
self.assertEqual(self.counter.result, {('17', '1'): 1,
('17', '2'): 3,
('17', '3'): 4
}
)
def test_str(self):
self.counter.get_stat_data()
self.assertEqual(str(self.counter), 'Contest #000017 Problem #1: 1 case\n'
'Contest #000017 Problem #2: 3 cases\n'
'Contest #000017 Problem #3: 4 cases\n')
def setUp(self):
self.problems = [Mock(cases=[i for i in range(1)], problem_id=('17', '1')),
Mock(cases=[i for i in range(3)], problem_id=('17', '2')),
Mock(cases=[i for i in range(4)], problem_id=('17', '3'))]
self.counter = CasesCounter(self.problems)
