1010from pages .logout .log_out_page import LogoutPage
1111from pages .datasets .subject_datasets_page import SubjectDatasetsPage
1212from pages .screening_subject_search .handover_into_symptomatic_care_page import (
13- HandoverIntoSymptomaticCarePage
13+ HandoverIntoSymptomaticCarePage ,
1414)
1515from utils .calendar_picker import CalendarPicker
1616from datetime import datetime
1717from pages .screening_subject_search .record_diagnosis_date_page import (
1818 RecordDiagnosisDatePage ,
1919)
2020from pages .screening_subject_search .diagnostic_test_outcome_page import (
21- DiagnosticTestOutcomePage ,OutcomeOfDiagnosticTest
21+ DiagnosticTestOutcomePage ,
22+ OutcomeOfDiagnosticTest ,
2223)
2324from pages .datasets .investigation_dataset_page import (
2425 InvestigationDatasetsPage ,
@@ -220,9 +221,12 @@ def after_high_risk_result(page: Page) -> None:
220221
221222 # The following code is on the diagnostic test outcome page
222223 DiagnosticTestOutcomePage (page ).verify_diagnostic_test_outcome ("High-risk findings" )
223- DiagnosticTestOutcomePage (page ).select_test_outcome_option (OutcomeOfDiagnosticTest .REFER_SURVEILLANCE )
224+ DiagnosticTestOutcomePage (page ).select_test_outcome_option (
225+ OutcomeOfDiagnosticTest .REFER_SURVEILLANCE
226+ )
224227 DiagnosticTestOutcomePage (page ).click_save_button ()
225228
229+
226230def after_lnpcp_result (page : Page ) -> None :
227231 InvestigationDatasetsPage (page ).expect_text_to_be_visible ("LNPCP" )
228232 BasePage (page ).click_back_button ()
@@ -238,9 +242,12 @@ def after_lnpcp_result(page: Page) -> None:
238242
239243 # The following code is on the diagnostic test outcome page
240244 DiagnosticTestOutcomePage (page ).verify_diagnostic_test_outcome ("LNPCP" )
241- DiagnosticTestOutcomePage (page ).select_test_outcome_option (OutcomeOfDiagnosticTest .REFER_SURVEILLANCE )
245+ DiagnosticTestOutcomePage (page ).select_test_outcome_option (
246+ OutcomeOfDiagnosticTest .REFER_SURVEILLANCE
247+ )
242248 DiagnosticTestOutcomePage (page ).click_save_button ()
243249
250+
244251def handover_subject_to_symptomatic_care (page : Page ) -> None :
245252 SubjectScreeningSummaryPage (page ).verify_latest_event_status_value (
246253 "A394 - Handover into Symptomatic Care for Surveillance - Patient Age"
@@ -287,7 +294,7 @@ def test_compartment_6(page: Page, smokescreen_properties: dict) -> None:
287294
288295 # The following code is on the investigation datasets page
289296 default_investigation_dataset_forms (page )
290- InvestigationDatasetsPage (page ).select_theraputic_procedure_type ()
297+ InvestigationDatasetsPage (page ).select_therapeutic_procedure_type ()
291298 default_investigation_dataset_forms_continuation (page )
292299 investigation_datasets_failure_reason (page )
293300 polyps_for_high_risk_result (page )
@@ -302,7 +309,7 @@ def test_compartment_6(page: Page, smokescreen_properties: dict) -> None:
302309
303310 # The following code is on the investigation datasets page
304311 default_investigation_dataset_forms (page )
305- InvestigationDatasetsPage (page ).select_theraputic_procedure_type ()
312+ InvestigationDatasetsPage (page ).select_therapeutic_procedure_type ()
306313 default_investigation_dataset_forms_continuation (page )
307314 investigation_datasets_failure_reason (page )
308315 polyps_for_high_risk_result (page )
@@ -332,7 +339,7 @@ def test_compartment_6(page: Page, smokescreen_properties: dict) -> None:
332339
333340 # The following code is on the investigation datasets page
334341 default_investigation_dataset_forms (page )
335- InvestigationDatasetsPage (page ).select_theraputic_procedure_type ()
342+ InvestigationDatasetsPage (page ).select_therapeutic_procedure_type ()
336343 default_investigation_dataset_forms_continuation (page )
337344 investigation_datasets_failure_reason (page )
338345 polyps_for_lnpcp_result (page )
@@ -347,7 +354,7 @@ def test_compartment_6(page: Page, smokescreen_properties: dict) -> None:
347354
348355 # The following code is on the investigation datasets page
349356 default_investigation_dataset_forms (page )
350- InvestigationDatasetsPage (page ).select_theraputic_procedure_type ()
357+ InvestigationDatasetsPage (page ).select_therapeutic_procedure_type ()
351358 default_investigation_dataset_forms_continuation (page )
352359 investigation_datasets_failure_reason (page )
353360 polyps_for_lnpcp_result (page )
@@ -400,7 +407,9 @@ def test_compartment_6(page: Page, smokescreen_properties: dict) -> None:
400407 DiagnosticTestOutcomePage (page ).verify_diagnostic_test_outcome (
401408 "Normal (No Abnormalities"
402409 )
403- DiagnosticTestOutcomePage (page ).select_test_outcome_option (OutcomeOfDiagnosticTest .INVESTIGATION_COMPLETE )
410+ DiagnosticTestOutcomePage (page ).select_test_outcome_option (
411+ OutcomeOfDiagnosticTest .INVESTIGATION_COMPLETE
412+ )
404413 DiagnosticTestOutcomePage (page ).click_save_button ()
405414
406415 SubjectScreeningSummaryPage (page ).verify_latest_event_status_value (
0 commit comments