reorganised the unit tests

acycliq · Oct 28, 2024 · 1f65dac · 1f65dac
1 parent 74bd898
commit 1f65dac
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/constants.py b/tests/constants.py
@@ -0,0 +1,16 @@
+# Test data constants
+EXPECTED_AREA_METRICS = {
+    'area_sum': 3721912.0,
+    'max_label': 3481,
+    'column_sums': [231498829.0, 150952363.0, 75868924.0, 151411232.0, 82555688.0]
+}
+
+# VarBayes iteration deltas
+EXPECTED_ITER_DELTAS = [
+    1.0, 0.8998741, 0.6707933, 0.4703735, 0.62746835, 0.58092755,
+    0.604991, 0.28188822, 0.26233155, 0.19196033, 0.5576391, 0.19038722,
+    0.24835205, 0.28241628, 0.06873486, 0.16072749, 0.5829934, 0.16633023,
+    0.14863184, 0.12343879, 0.099603266, 0.05200758, 0.06741488, 0.1034664,
+    0.18765935, 0.16917473, 0.06632929, 0.046282567, 0.033078548, 0.020875283,
+    0.012404523
+]