Skip to content

Commit 2c04f38

Browse files
committed
fix metacluster bug with test
1 parent 05c602f commit 2c04f38

2 files changed

Lines changed: 8 additions & 1 deletion

File tree

src/flowsom/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ def metacluster(self, n_clusters=None):
288288
"""
289289
if n_clusters is None:
290290
n_clusters = self.n_clusters
291-
self.model.set_n_clusters(n_clusters)
291+
self.model.set_n_clusters(n_clusters)
292292
self.model.metacluster_model.fit_predict(self.model.codes)
293293
return self
294294

tests/models/test_FlowSOMModel.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from sklearn.metrics import v_measure_score
22

3+
from flowsom.main import FlowSOM
34
from flowsom.models import FlowSOMEstimator
45

56

@@ -33,3 +34,9 @@ def test_reproducibility_seed(X):
3334
y_pred_2 = fsom_2.fit_predict(X)
3435

3536
assert all(y_pred_1 == y_pred_2)
37+
38+
def test_metacluster(X):
39+
fsom = FlowSOM(X, n_clusters=10)
40+
assert 10 == fsom.model.metacluster_model.n_clusters
41+
fsom.metacluster(n_clusters=5)
42+
assert 5 == fsom.model.metacluster_model.n_clusters

0 commit comments

Comments
 (0)