def test_get_problem_meta(self): """ Test get_problem_meta_for helper function. """ problem_meta = get_problem_meta_for(self.participant) meta = [] for problem in self.participant.competition.problem_set.all(): new_dict = _get_problem_dict_for(problem) meta.append(new_dict) self.assertEquals(len(problem_meta), len(meta))
def test_get_problem_dict_for(self): """ Test _get_problem_dict_for() method. """ problems = self.participant.competition.problem_set.all() for problem in problems: problem_dict = _get_problem_dict_for(problem) self.assertEquals( len(problem_dict["submission"]), len(problem.submission_set.filter(participant=self.participant, problem=problem)), ) self.assertEquals(problem_dict["problem"], problem)