wip

Author: Andyg79

pages/datasets/investigation_dataset_page.py

@@ -32,6 +32,7 @@ def __init__(self, page: Page):
         self.show_contrast_tagging_and_drug_information = self.page.locator(
             "#anchorContrastTaggingDrug"
         )
+        self.show_radiology_information = self.page.locator("#anchorRadiology")
         self.aspirant_endoscopist_link = self.page.locator(
             "#UI_ASPIRANT_ENDOSCOPIST_PIO_SELECT_LINK"
         )
@@ -202,6 +203,13 @@ def click_show_contrast_tagging_and_drug_information(self) -> None:
         """
         self.click(self.show_contrast_tagging_and_drug_information)
 
+    def click_show_radiology_information(self) -> None:
+        """
+        This method is designed to click on the show radiology information link.
+        It clicks on the show radiology information link.
+        """
+        self.click(self.show_radiology_information)
+
     def select_aspirant_endoscopist_option(self, option: str) -> None:
         """
         This method is designed to select an aspirant endoscopist from the aspirant endoscopist options.
@@ -1187,7 +1195,7 @@ class DrugTypeOptions(StrEnum):
     MOVIPREP = "200536~Sachet(s)"
     CITRAMAG = "200538~Sachet(s)"
     MANNITOL = "200539~Litre(s)"
-    GASTROGRAFIN = "200540~Mls Solution"
+    GASTROGRAFIN = "200540~Mls Solution~204334"
     PHOSPHATE_ENEMA = "200528~Sachet(s)"
     MICROLAX_ENEMA = "200529~Sachet(s)"
     OSMOSPREP = "203063~Tablet(s)"

utils/investigation_dataset.py

@@ -340,16 +340,23 @@ def complete_dataset_with_args(
         extracolonic_summary_code: Optional[str] = None,
         intracolonic_summary_code: Optional[str] = None,
     ) -> None:
-        """This method completes the investigation dataset with the provided dictionaries.
+        """
+        Completes the investigation dataset using the provided dictionaries for each section.
+
         Args:
-            general_information (dict): A dictionary containing the general information to be filled in the form.
-            drug_information (dict): A dictionary containing the drug information to be filled in the form.
-            endoscopy_information (dict): A dictionary containing the endoscopy information to be filled in the form.
-            failure_information (dict): A dictionary containing the failure information to be filled in the form.
-            completion_information (Optional[dict]): An optional  dictionary containing the completion information to be filled in the form.
-            polyp_information (Optional[list]): An optional list containing the polyp information to be filled in the form.
-            polyp_intervention (Optional[list]): An optional list containing the polyp intervention to be filled in the form.
-            polyp_histology (Optional[list]): An optional list containing the polyp histology to be filled in the form.
+            general_information (Optional[dict]): General dataset fields (site, practitioner, clinician, etc.).
+            drug_information (Optional[dict]): Drug types and dosages for the main drug section.
+            endoscopy_information (Optional[dict]): Endoscopy procedure details and related fields.
+            failure_information (Optional[dict]): Failure reasons and related information.
+            completion_information (Optional[dict]): Completion proof parameters.
+            polyp_information (Optional[list[dict]]): List of polyp information dictionaries.
+            polyp_intervention (Optional[list[dict] or list[list[dict]]]): List of interventions per polyp, or lists of interventions for each polyp.
+            polyp_histology (Optional[list[dict]]): List of polyp histology dictionaries.
+            contrast_tagging_and_drug (Optional[dict]): Contrast, tagging agent, and drug information.
+            tagging_agent_given_drug_information (Optional[dict]): Tagging agent drug types and doses.
+            radiology_information (Optional[dict]): Radiology section fields.
+            extracolonic_summary_code (Optional[str]): Extracolonic summary code value.
+            intracolonic_summary_code (Optional[str]): Intracolonic summary code value.
         """
         logging.info("Completing investigation dataset with the provided dictionaries")
         # Investigation Dataset
@@ -615,6 +622,8 @@ def fill_out_radiology_information(self, radiology_data: dict) -> None:
         """
         logging.info("Filling out Radiology Information")
 
+        self.investigation_datasets_pom.click_show_radiology_information()
+
         # Define mapping for each radiology field and its selector
         radiology_map = {
             "examination_quality": "#UI_EXAM_QUALITY",