@@ -28,10 +28,15 @@ def __init__(self, page: Page):
2828 self .record_other_post_investigation_contact_button = self .page .get_by_role (
2929 "button" , name = "Record other post-"
3030 )
31- self .enter_diagnostic_test_outcome_button = self .page .get_by_role ("button" , name = "Enter Diagnostic Test Outcome"
31+ self .enter_diagnostic_test_outcome_button = self .page .get_by_role (
32+ "button" , name = "Enter Diagnostic Test Outcome"
33+ )
34+ self .handover_into_symptomatic_care_button = page .get_by_role (
35+ "button" , name = "Handover into Symptomatic Care"
36+ )
37+ self .record_diagnosis_date_button = page .get_by_role (
38+ "button" , name = "Record Diagnosis Date"
3239 )
33- self .handover_into_symptomatic_care_button = page .get_by_role ("button" , name = "Handover into Symptomatic Care" )
34- self .record_diagnosis_date_button = page .get_by_role ("button" , name = "Record Diagnosis Date" )
3540
3641 def click_suitable_for_endoscopic_test_button (self ) -> None :
3742 """Click the 'Suitable for Endoscopic Test' button."""
@@ -85,12 +90,12 @@ def click_record_other_post_investigation_contact_button(self) -> None:
8590
8691 def click_enter_diagnostic_test_outcome_button (self ) -> None :
8792 """Click the 'Enter Diagnostic Test Outcome' button."""
88- self .click (self .enter_diagnostic_test_outcome_button )
93+ self .click (self .enter_diagnostic_test_outcome_button )
8994
9095 def click_handover_into_symptomatic_care_button (self ) -> None :
9196 """Click the 'Handover Into Symptomatic Care' button."""
92- self .click (self .handover_into_symptomatic_care_button )
97+ self .click (self .handover_into_symptomatic_care_button )
9398
9499 def click_record_diagnosis_date_button (self ) -> None :
95100 """Click the 'Record Diagnosis Date' button."""
96- self .click (self .record_diagnosis_date_button )
101+ self .click (self .record_diagnosis_date_button )
0 commit comments