tests: add new unitary tests

Author: judithspd

anjana/anonymity/utils/utils.py

@@ -222,6 +222,8 @@ def apply_transformation(
     for i, qi in enumerate(quasi_ident):
         hierarchy_qi = hierarchies[qi]
         level = transformation[i]
+        if level < 0:
+            raise ValueError("Error, invalid hierarchy level")
         if level > max(hierarchies[qi].keys()):
             raise ValueError("Error, invalid hierarchy level")
         actual = actual_transform[qi]

tests/test_unitary.py

@@ -1,6 +1,7 @@
 import unittest
 import anjana
 from anjana import anonymity
+from anjana.anonymity import utils
 import pandas as pd
 import beartype
 
@@ -556,3 +557,15 @@ def test_return_delta_dic(self):
             self.hierarchies,
         )
         assert isinstance(data_anon, pd.DataFrame)
+
+    def test_apply_transformation_neg(self):
+        with self.assertRaises(ValueError):
+            utils.apply_transformation(
+                self.data, self.quasi_ident, self.hierarchies, [-1, 1, 1]
+            )
+
+    def test_apply_transformation_out(self):
+        with self.assertRaises(ValueError):
+            utils.apply_transformation(
+                self.data, self.quasi_ident, self.hierarchies, [100, 1, 1]
+            )