Check using expected values

talmolab · Sep 29, 2023 · 2c8fc45 · 2c8fc45
1 parent f1b0be4
commit 2c8fc45
Showing 1 changed file with 27 additions and 30 deletions.
diff --git a/tests/info/test_metrics.py b/tests/info/test_metrics.py
@@ -23,36 +23,33 @@ def test_matched_instance_distances(centered_pair_labels, centered_pair_predicti
 
     # Show mean difference for each node
     node_names = labels_gt.skeletons[0].node_names
+    expected_values = {
+        "head": 0.872426920709296,
+        "neck": 0.8016280746914615,
+        "thorax": 0.8602021363390538,
+        "abdomen": 1.01012200038258,
+        "wingL": 1.1297727023475939,
+        "wingR": 1.0869857897008424,
+        "forelegL1": 0.780584225081443,
+        "forelegL2": 1.170805798894702,
+        "forelegL3": 1.1020486509389473,
+        "forelegR1": 0.9014698776116817,
+        "forelegR2": 0.9448001033112047,
+        "forelegR3": 1.308385214215777,
+        "midlegL1": 0.9095691623265347,
+        "midlegL2": 1.2203595627907582,
+        "midlegL3": 0.9813843358470163,
+        "midlegR1": 0.9871017182813739,
+        "midlegR2": 1.0209829335569256,
+        "midlegR3": 1.0990681234096988,
+        "hindlegL1": 1.0005335192834348,
+        "hindlegL2": 1.273539518539708,
+        "hindlegL3": 1.1752245985832817,
+        "hindlegR1": 1.1402833959265248,
+        "hindlegR2": 1.3143221301212737,
+        "hindlegR3": 1.0441458592503365,
+    }
 
     for node_idx, node_name in enumerate(node_names):
         mean_d = np.nanmean(D[..., node_idx])
-        # print(f"{node_name}\t\t{mean_d}")
-
-        """Expected values (instance-wise matching):
-
-        head            0.872426920709296
-        neck            0.8016280746914615
-        thorax          0.8602021363390538
-        abdomen         1.01012200038258
-        wingL           1.1297727023475939
-        wingR           1.0869857897008424
-        forelegL1               0.780584225081443
-        forelegL2               1.170805798894702
-        forelegL3               1.1020486509389473
-        forelegR1               0.9014698776116817
-        forelegR2               0.9448001033112047
-        forelegR3               1.308385214215777
-        midlegL1                0.9095691623265347
-        midlegL2                1.2203595627907582
-        midlegL3                0.9813843358470163
-        midlegR1                0.9871017182813739
-        midlegR2                1.0209829335569256
-        midlegR3                1.0990681234096988
-        hindlegL1               1.0005335192834348
-        hindlegL2               1.273539518539708
-        hindlegL3               1.1752245985832817
-        hindlegR1               1.1402833959265248
-        hindlegR2               1.3143221301212737
-        hindlegR3               1.0441458592503365"""
-        assert mean_d > 0.77
-        assert mean_d < 1.32
+        assert np.isclose(mean_d, expected_values[node_name], atol=1e-6)