def test_query_for_problem(self):
     self.assertEqual(
         2,
         len(DjangoGateway._query_problem_logs(problem_ids=[self.prob.pk])))
 def test_query_all(self):
     self.assertEqual(3, len(DjangoGateway._query_problem_logs()))