⬆️ unittest

Author: GiulioRossetti

ash_model/test/test_s_centralities.py

@@ -16,41 +16,41 @@ def get_hypergraph():
 
     def test_centralities(self):
         a = self.get_hypergraph()
-        self.assertDictEqual(
+        _almost_equal(
             s_betweenness_centrality(a, s=1),
             {"e1": 0.0, "e2": 0.0, "e3": 0.0, "e4": 0.0, "e5": 0.0},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_betweenness_centrality(a, s=1, edges=False),
             {1: 0.0, 2: 0.0, 3: 0.0, 4: 0.0},
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_closeness_centrality(a, s=1),
             {"e1": 1.0, "e2": 1.0, "e3": 1.0, "e4": 1.0, "e5": 1.0},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_closeness_centrality(a, s=1, edges=False),
             {1: 1.0, 2: 1.0, 3: 1.0, 4: 1.0},
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_eccentricity(a, s=1),
             {"e1": 1.0, "e2": 1.0, "e3": 1.0, "e4": 1.0, "e5": 1.0},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_eccentricity(a, s=1, edges=False), {1: 1, 2: 1, 3: 1, 4: 1}
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_harmonic_centrality(a, s=1),
             {"e1": 4.0, "e3": 4.0, "e5": 4.0, "e4": 4.0, "e2": 4.0},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_harmonic_centrality(a, s=1, edges=False), {1: 3.0, 2: 3.0, 3: 3.0, 4: 3.0}
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_katz(a, s=1),
             {
                 "e1": 0.447213595499958,
@@ -60,20 +60,20 @@ def test_centralities(self):
                 "e5": 0.4472135954999579,
             },
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_katz(a, s=1, edges=False),
             {1: 0.49999999999999983, 2: 0.5, 3: 0.5, 4: 0.49999999999999994},
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_load_centrality(a, s=1),
             {"e1": 0.0, "e2": 0.0, "e3": 0.0, "e4": 0.0, "e5": 0.0},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_load_centrality(a, s=1, edges=False), {1: 0.0, 2: 0.0, 3: 0.0, 4: 0.0}
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_eigenvector_centrality(a, s=1),
             {
                 "e1": 0.447213595499958,
@@ -83,7 +83,7 @@ def test_centralities(self):
                 "e5": 0.4472135954999579,
             },
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_eigenvector_centrality(a, s=1, edges=False),
             {
                 1: 0.4999999999999999,
@@ -93,11 +93,11 @@ def test_centralities(self):
             },
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_information_centrality(a, s=1),
             {"e1": 0.625, "e2": 0.625, "e3": 0.625, "e4": 0.625, "e5": 0.625},
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_information_centrality(a, s=1, edges=False),
             {
                 1: 0.6666666666666666,
@@ -107,7 +107,7 @@ def test_centralities(self):
             },
         )
 
-        self.assertDictEqual(
+        _almost_equal(
             s_second_order_centrality(a, s=1),
             {
                 "e1": 3.4641016151377535,
@@ -117,7 +117,7 @@ def test_centralities(self):
                 "e5": 3.4641016151377544,
             },
         )
-        self.assertDictEqual(
+        _almost_equal(
             s_second_order_centrality(a, s=1, edges=False),
             {
                 1: 2.4494897427831774,
@@ -126,3 +126,16 @@ def test_centralities(self):
                 4: 2.4494897427831774,
             },
         )
+
+        _almost_equal(s_second_order_centrality(a, s=1, edges=False), {
+            1: 2.4494897427831774,
+            2: 2.4494897427831774,
+            3: 2.4494897427831774,
+            4: 2.4494897427831774,
+        })
+
+
+def _almost_equal(A, B):
+    for k in A:
+        np.testing.assert_almost_equal(A[k], B[k])
+