add base tests

nitrain · May 4, 2024 · e09880c · e09880c
1 parent 63fc37c
commit e09880c
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/test_transforms.py b/tests/test_transforms.py
@@ -301,6 +301,27 @@ def test_Translate(self):
         img2d_tx = my_tx(img2d)
 
         img3d_tx = my_tx(img3d)
+
+class TestErrors(unittest.TestCase):
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def test_base_transform(self):
+        from nitrain.transforms import BaseTransform
+        tx = BaseTransform()
+
+        with self.assertRaises(NotImplementedError):
+            tx.fit()
+
+        with self.assertRaises(NotImplementedError):
+            tx.__call__()
+
+        with self.assertRaises(NotImplementedError):
+            tx.__repr__()
 
 if __name__ == '__main__':
     run_tests()