fix FlowSOM labels attr

Author: berombau

src/flowsom/main.py

@@ -89,21 +89,37 @@ def __init__(
         self._update_derived_values()
 
     @property
-    def labels_(self):
-        """Get the labels."""
+    def cluster_labels(self):
+        """Get the cluster labels."""
         if "cell_data" in self.mudata.mod.keys():
             if "clustering" in self.mudata["cell_data"].obs_keys():
                 return self.mudata["cell_data"].obs["clustering"]
         return None
 
-    @labels_.setter
-    def labels_(self, value):
-        """Set the labels."""
+    @cluster_labels.setter
+    def cluster_labels(self, value):
+        """Set the cluster labels."""
         if "cell_data" in self.mudata.mod.keys():
             self.mudata["cell_data"].obs["clustering"] = value
         else:
             raise ValueError("No cell data found in the MuData object.")
 
+    @property
+    def metacluster_labels(self):
+        """Get the metacluster labels."""
+        if "cell_data" in self.mudata.mod.keys():
+            if "clustering" in self.mudata["cell_data"].obs_keys():
+                return self.mudata["cell_data"].obs["metaclustering"]
+        return None
+
+    @metacluster_labels.setter
+    def metacluster_labels(self, value):
+        """Set the metacluster labels."""
+        if "cell_data" in self.mudata.mod.keys():
+            self.mudata["cell_data"].obs["metaclustering"] = value
+        else:
+            raise ValueError("No cell data found in the MuData object.")
+
     def read_input(
         self,
         inp=None,

tests/test_basic.py

@@ -1,3 +1,5 @@
+import numpy as np
+
 import flowsom as fs
 
 
@@ -18,6 +20,14 @@ def test_FlowSOM_type(FlowSOM_res):
     assert isinstance(FlowSOM_res, fs.FlowSOM)
 
 
+def test_FlowSOM_cluster_labels(FlowSOM_res):
+    assert len(np.unique(FlowSOM_res.cluster_labels)) == 100
+
+
+def test_FlowSOM_metacluster_labels(FlowSOM_res):
+    assert len(np.unique(FlowSOM_res.metacluster_labels)) == 10
+
+
 def test_plot_stars(FlowSOM_res, tmp_path):
     pl = fs.pl.plot_stars(
         FlowSOM_res,