call from top-level module

ANTsX · Nov 18, 2024 · 3433a63 · 3433a63
1 parent 26fc83b
commit 3433a63
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1012,9 +1012,7 @@ def tearDown(self):
         pass
 
     def test_from_sitk(self):
-        from ants.utils.sitk_to_ants import from_sitk
-
-        ants_img = from_sitk(self.img)
+        ants_img = ants.from_sitk(self.img)
 
         with TemporaryDirectory() as temp_dir:
             temp_fpath = os.path.join(temp_dir, "img.nrrd")
@@ -1029,14 +1027,12 @@ def test_from_sitk(self):
         nptest.assert_almost_equal(self.img.GetDirection(), img.GetDirection())
 
     def test_ito_sitk(self):
-        from ants.utils.sitk_to_ants import to_sitk
-
         with TemporaryDirectory() as temp_dir:
             temp_fpath = os.path.join(temp_dir, "img.nrrd")
             sitk.WriteImage(self.img, temp_fpath)
             ants_img = ants.image_read(temp_fpath)
 
-        img = to_sitk(ants_img)
+        img = ants.to_sitk(ants_img)
 
         nptest.assert_equal(
             sitk.GetArrayViewFromImage(self.img), sitk.GetArrayViewFromImage(img)