working tests

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Mar 1, 2024 · 45f2458 · 45f2458
1 parent 8fadd54
commit 45f2458
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_registrators.py b/tests/test_registrators.py
@@ -6,7 +6,8 @@
 from auxiliary.turbopath import turbopath
 
 from brainles_preprocessing.registration.ANTs.ANTs import ANTsRegistrator
-from brainles_preprocessing.registration.niftyreg.niftyreg import NiftyRegRegistrator
+from brainles_preprocessing.registration.niftyreg.niftyreg import \
+    NiftyRegRegistrator
 
 
 class RegistratorBase:
@@ -31,7 +32,7 @@ def setUp(self):
         self.moving_image = input_dir + "/bet_tcia_example_t1c_mask.nii.gz"
 
         self.matrix = self.output_dir + "/matrix"
-        self.transform_matrix = f"{self.matrix}.{self.matrix_extension}"
+        self.transform_matrix = input_dir + f"/matrix.{self.matrix_extension}"
 
     def tearDown(self):
         # Clean up created files if they exist