diff --git a/rad/tests/test_helpers.py b/rad/tests/test_helpers.py index c31822a5..e16c029b 100644 --- a/rad/tests/test_helpers.py +++ b/rad/tests/test_helpers.py @@ -504,9 +504,7 @@ class ExamTester: cid_exam_scores_res = self.client.get(exam_scores_url) cid_exam_scores_soup = BeautifulSoup(cid_exam_scores_res.content, "html.parser") - submitted_answers = cid_exam_scores_soup.find_all( - "span", {"class": "submitted-user-answer"} - ) + submitted_answers = cid_exam_scores_soup.find_all(class_="submitted-user-answer") match self.app_name: case "physics": ols = cid_exam_scores_soup.find_all("ol", class_="mb-0") @@ -578,9 +576,7 @@ class ExamTester: ans = question.get_correct_answer() assert correct_answers[i].text.strip() == f"Correct answer: {ans}" - submitted_answers = cid_exam_scores_soup.find_all( - "span", {"class": "submitted-user-answer"} - ) + submitted_answers = cid_exam_scores_soup.find_all(class_="submitted-user-answer") assert len(submitted_answers) == self.answers_per_question * len( self.generated_questions )