fix: missing pytest import and passing flattened SEG to alignImages

bhklab · Dec 8, 2023 · bab47e4 · bab47e4
1 parent b38e7a4
commit bab47e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_image_processing.py b/tests/test_image_processing.py
@@ -1,5 +1,6 @@
 from yarea.loaders import *
 from yarea.image_processing import *
+import pytest
 
 @pytest.fixture
 def nsclcCTImage():
@@ -24,7 +25,7 @@ def test_flattenImage(nsclcSEGImage):
 def test_alignImages(nsclcCTImage, nsclcSEGImage):
     """Test setting segmentation origin, direction, and spacing to match the CT after flattening"""
     flattenedSEG = flattenImage(nsclcSEGImage)
-    actual = alignImages(nsclcCTImage, nsclcSEGImage)
+    actual = alignImages(nsclcCTImage, flattenedSEG)
     assert actual.GetSize() == (512, 512, 304), \
         "Wrong image size"
     assert actual.GetSpacing() == (0.693359375, 0.693359375, 1.0), \