Fix emulator view position bug for Implant Displaced

src/modality_emulator.py

@@ -61,7 +61,9 @@ def generate_dicom(self) -> Dataset:
             logger.error("No dataset provided for DICOM generation")
             return ds
 
-        img_path = f"{SAMPLE_IMAGES_PATH}/L{self.view.replace('ID', '')}.jpg"
+        view_position = self.view.replace("ID", "")
+
+        img_path = f"{SAMPLE_IMAGES_PATH}/L{view_position}.jpg"
         img = Image.open(img_path).convert("L")
         if self.laterality == "R":
             img = img.transpose(Image.Transpose.FLIP_LEFT_RIGHT)
@@ -89,7 +91,7 @@ def generate_dicom(self) -> Dataset:
         ds.PixelRepresentation = 0
         ds.PixelData = pixel_bytes
         ds.ImageLaterality = self.laterality
-        ds.ViewPosition = self.view
+        ds.ViewPosition = view_position
 
         ds.AccessionNumber = self.dataset.AccessionNumber
         ds.PatientID = self.dataset.PatientID

tests/scripts/test_database.py

@@ -1,8 +1,9 @@
-import pytest
 import sqlite3
 import sys
 from pathlib import Path
 
+import pytest
+
 sys.path.append(f"{Path(__file__).parent.parent.parent}/scripts/python")
 
 from database import backup_database, reset_worklist_database

tests/test_modality_emulator.py

@@ -57,7 +57,7 @@ def test_generate_dicom_creates_valid_dataset(
             dicom = DicomExample(
                 dataset=ds,
                 laterality="L",
-                view="CC",
+                view="CCID",
                 study_instance_uid=study_instance_uid,
                 series_number=1,
             )