registrator tests

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Mar 1, 2024 · d0ad94d · d0ad94d
1 parent 13036c4
commit d0ad94d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_niftyreg_registrator.py → tests/test_registrators.py b/tests/test_niftyreg_registrator.py → tests/test_registrators.py
@@ -21,7 +21,7 @@ def setUp(self):
         self.output_dir = test_data_dir + "/temp_output_niftyreg"
         os.makedirs(self.output_dir, exist_ok=True)
 
-        self.registrator = get_registrator()
+        self.registrator = self.get_registrator()
 
         self.fixed_image = input_dir + "/tcia_example_t1c.nii.gz"
         self.moving_image = input_dir + "/bet_tcia_example_t1c_mask.nii.gz"
@@ -66,7 +66,6 @@ class TestANTsRegistratorBase(TestRegistratorBase):
     def get_registrator(self):
         return ANTsRegistrator()
 
-
 class TestNiftyRegRegistratorRegistratorBase(TestRegistratorBase):
     def get_registrator(self):
         return NiftyRegRegistrator()