Update unittests.py

nla-group · Apr 6, 2024 · 71b8e69 · 71b8e69
1 parent f6d0b8a
commit 71b8e69
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/unittests.py b/unittests.py
@@ -361,6 +361,18 @@ def test_aggregate_precompute(self):
 
             _, _, _, _, _, _, _ = aggregate_cm.general_aggregate(data, sorting="NA", tol=0.5)
             _, _, _, _, _, _, _ = aggregate_c.general_aggregate(data, "NA", 0.5)
+
+            inverse_ind1, spl1, _, _, _, _, _ = aggregate.lm_aggregate(data, sorting="pca", tol=0.5)
+            inverse_ind2, spl2, _, _, _, _, _ = aggregate_cm.lm_aggregate(data, sorting="pca", tol=0.5)
+            inverse_ind3, spl3, _, _, _, _, _ = aggregate_c.lm_aggregate(data, "pca", 0.5)
+
+            _, _, _, _, _, _, _ = aggregate.lm_aggregate(data, sorting="norm-mean", tol=0.5)
+            _, _, _, _, _, _, _ = aggregate_cm.lm_aggregate(data, sorting="norm-mean", tol=0.5)
+            _, _, _, _, _, _, _ = aggregate_c.lm_aggregate(data, "norm-mean", 0.5)
+
+            _, _, _, _, _, _, _ = aggregate.lm_aggregate(data, sorting="NA", tol=0.5)
+            _, _, _, _, _, _, _ = aggregate_cm.lm_aggregate(data, sorting="NA", tol=0.5)
+            _, _, _, _, _, _, _ = aggregate_c.lm_aggregate(data, "NA", 0.5)
 
             if np.sum(inverse_ind1 != inverse_ind2) != 0:
                 checkpoint = 0