diff --git a/src/modality_emulator.py b/src/modality_emulator.py index 1517a1f..398bbd3 100644 --- a/src/modality_emulator.py +++ b/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 diff --git a/tests/scripts/test_database.py b/tests/scripts/test_database.py index 3d73b6e..8c1990d 100644 --- a/tests/scripts/test_database.py +++ b/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 diff --git a/tests/test_modality_emulator.py b/tests/test_modality_emulator.py index 3916009..4dce4b8 100644 --- a/tests/test_modality_emulator.py +++ b/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, )