fixed up test dicom test

Australian-Imaging-Service · Sep 27, 2024 · b701899 · b701899
1 parent f5ecd50
commit b701899
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_dicom.py b/tests/test_dicom.py
@@ -8,9 +8,9 @@ def test_dicom_creation(dicom_module):
     out_dir = Path(tempfile.mkdtemp())
 
     # Generate DICOM files
-    dicom_module.get_image(out_dir)
+    dicom_dir = dicom_module.get_image(out_dir)
 
-    assert len(list(out_dir.glob('*.dcm'))) == dicom_module.num_vols
+    assert len(list(dicom_dir.glob("*.dcm"))) == dicom_module.num_vols
 
-    for p in out_dir.glob('*.dcm'):
+    for p in dicom_dir.glob("*.dcm"):
         pydicom.dcmread(p)  # Check dicom file can be read