class TestProblemsTree(unittest.TestCase):
def setUp(self):
self.problems = []
self.problems.append(Mock(cases=['a1', 'a2', 'a3', 'a4', 'a5', 'a6'], problem_id=('42', '1')))
self.problems[-1].name = '42a'
self.problems.append(Mock(cases=['b1', 'b2', 'b3', 'b4', 'b5'], problem_id=('42', '2')))
self.problems[-1].name = '42b'
self.problems.append(Mock(cases=['b1', 'b2', 'b3', 'b4', 'b5', 'b6', 'b7'], problem_id=('43', '2')))
self.problems[-1].name = '43a'
self.problems.append(Mock(cases=['a1', 'a2', 'a3', 'a4', 'a5', 'a7'], problem_id=('43', '1')))
self.problems[-1].name = '43b'
self.problems.append(Mock(cases=['c1', 'c2', 'c3', 'c4'], problem_id=('44', '1')))
self.problems[-1].name = '44a'
self.problems.append(Mock(cases=['c1', 'c2', 'c3'], problem_id=('45', '1')))
self.problems[-1].name = '45a'
self.problems.append(Mock(cases=['c1', 'c2', 'c3', 'c4'], problem_id=('46', '1')))
self.problems[-1].name = '46a'
for problem in self.problems:
string = 'Problem #{0} ("{1}") from contest #{2}'. \
format(problem.problem_id[1], problem.name, problem.problem_id[0])
problem.__str__ = Mock(return_value=string)
self.tree = ProblemsTree(self.problems)
def test_common(self):
self.assertEqual(self.tree.get_previous_problem(self.problems[0]), None)
self.assertEqual(self.tree.get_previous_problem(self.problems[1]), None)
self.assertEqual(self.tree.get_previous_problem(self.problems[2]), self.problems[1])
self.assertEqual(self.tree.get_previous_problem(self.problems[3]), self.problems[0])
self.assertEqual(self.tree.get_previous_problem(self.problems[4]), None)
self.assertEqual(self.tree.get_previous_problem(self.problems[5]), self.problems[4])
self.assertEqual(self.tree.get_previous_problem(self.problems[6]), self.problems[4])
def test_str(self):
result_string = str(self.tree)
correct_string = """Problem #1 ("42a") from contest #42: it is a new problem. Tests: 6.
Problem #2 ("42b") from contest #42: it is a new problem. Tests: 5.
Problem #2 ("43a") from contest #43: is based on Problem #2 ("42b") from contest #42. Tests: +2, -0, 5 not changed.
Problem #1 ("43b") from contest #43: is based on Problem #1 ("42a") from contest #42. Tests: +1, -1, 5 not changed.
Problem #1 ("44a") from contest #44: it is a new problem. Tests: 4.
Problem #1 ("45a") from contest #45: is based on Problem #1 ("44a") from contest #44. Tests: +0, -1, 3 not changed.
Problem #1 ("46a") from contest #46: it is Problem #1 ("44a") from contest #44. Tests: 4.
"""
self.assertEqual(result_string, correct_string)
def test_get_problems(self):
self.tree.problems = 42
self.assertEqual(self.tree.get_problems(), 42)
def test_parent_relation(self):
tree = ProblemsTree(self.problems)
problem = Mock()
tree.problem_previous[problem] = (1, 2, 3, 4, 5)
self.assertEqual(tree.get_relation_to_parent(problem), (1, 2, 3, 4, 5))
def setUp(self):
self.problems = []
self.problems.append(Mock(cases=['a1', 'a2', 'a3', 'a4', 'a5', 'a6'], problem_id=('42', '1')))
self.problems[-1].name = '42a'
self.problems.append(Mock(cases=['b1', 'b2', 'b3', 'b4', 'b5'], problem_id=('42', '2')))
self.problems[-1].name = '42b'
self.problems.append(Mock(cases=['b1', 'b2', 'b3', 'b4', 'b5', 'b6', 'b7'], problem_id=('43', '2')))
self.problems[-1].name = '43a'
self.problems.append(Mock(cases=['a1', 'a2', 'a3', 'a4', 'a5', 'a7'], problem_id=('43', '1')))
self.problems[-1].name = '43b'
self.problems.append(Mock(cases=['c1', 'c2', 'c3', 'c4'], problem_id=('44', '1')))
self.problems[-1].name = '44a'
self.problems.append(Mock(cases=['c1', 'c2', 'c3'], problem_id=('45', '1')))
self.problems[-1].name = '45a'
self.problems.append(Mock(cases=['c1', 'c2', 'c3', 'c4'], problem_id=('46', '1')))
self.problems[-1].name = '46a'
for problem in self.problems:
string = 'Problem #{0} ("{1}") from contest #{2}'. \
format(problem.problem_id[1], problem.name, problem.problem_id[0])
problem.__str__ = Mock(return_value=string)
self.tree = ProblemsTree(self.problems)
def test_parent_relation(self):
tree = ProblemsTree(self.problems)
problem = Mock()
tree.problem_previous[problem] = (1, 2, 3, 4, 5)
self.assertEqual(tree.get_relation_to_parent(problem), (1, 2, 3, 4, 5))
