chore(test): use unittest assertions in test_metrics.py

aequitas-aod · Dec 14, 2023 · ed273ea · ed273ea
1 parent a7f2235
commit ed273ea
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/test/core/test_metrics.py b/test/core/test_metrics.py
@@ -11,18 +11,23 @@ def setUp(self) -> None:
         self.fair_dataset = uniform_binary_dataset(rows=DATASET_SIZE)
         self.unfair_dataset = skewed_binary_dataset(rows=DATASET_SIZE, p=0.9)
 
+    def assertInRange(self, value, lower, upper):
+        self.assertGreaterEqual(value, lower)
+        self.assertLessEqual(value, upper)
+
     def test_parity_on_fair_binary_case(self):
         x = self.fair_dataset[:, 0]
         y = self.fair_dataset[:, 1]
         parities = discrete_demographic_parities(x, y, 1)
-        assert parities.shape == (1,)
-        assert 0.0 < parities[0] <= 0.005
+        self.assertEqual(parities.shape, (1,))
+        self.assertInRange(parities[0], 0.0, 0.005)
 
     def test_parity_on_unfair_binary_case(self):
         x = self.unfair_dataset[:, 0]
         y = self.unfair_dataset[:, 1]
         parities = discrete_demographic_parities(x, y, 1)
-        assert 0.4 < parities[0] <= 0.5
+        self.assertEqual(parities.shape, (1,))
+        self.assertInRange(parities[0], 0.4, 0.5)
 
 
 if __name__ == '__main__':