fix metacluster bug with test

saeyslab · Nov 25, 2024 · 2c04f38 · 2c04f38
1 parent 05c602f
commit 2c04f38
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/flowsom/main.py b/src/flowsom/main.py
@@ -288,7 +288,7 @@ def metacluster(self, n_clusters=None):
         """
         if n_clusters is None:
             n_clusters = self.n_clusters
-            self.model.set_n_clusters(n_clusters)
+        self.model.set_n_clusters(n_clusters)
         self.model.metacluster_model.fit_predict(self.model.codes)
         return self
 

diff --git a/tests/models/test_FlowSOMModel.py b/tests/models/test_FlowSOMModel.py
@@ -1,5 +1,6 @@
 from sklearn.metrics import v_measure_score
 
+from flowsom.main import FlowSOM
 from flowsom.models import FlowSOMEstimator
 
 
@@ -33,3 +34,9 @@ def test_reproducibility_seed(X):
     y_pred_2 = fsom_2.fit_predict(X)
 
     assert all(y_pred_1 == y_pred_2)
+
+def test_metacluster(X):
+    fsom = FlowSOM(X, n_clusters=10)
+    assert 10 == fsom.model.metacluster_model.n_clusters
+    fsom.metacluster(n_clusters=5)
+    assert 5 == fsom.model.metacluster_model.n_clusters