add test for network

Author: AntSimi

src/py_eddy_tracker/observations/network.py

@@ -1104,14 +1104,28 @@ def get_group_array(self, results, nb_obs):
                     if g0 > g1:
                         g0, g1 = g1, g0
                     merge_id.append((g0, g1))
+        gr_transfer = self.group_translator(id_free, set(merge_id))
+        return gr_transfer[gr]
+
+    @staticmethod
+    def group_translator(nb, duos):
+        """
+        Create a translator with all duos
 
-        # FIXME: how it's work when several merge ? like (0,1), (0,2), (1,3)
-        gr_transfer = arange(id_free, dtype="u4")
-        for i, j in set(merge_id):
-            gr_i, gr_j = gr_transfer[i], gr_transfer[j]
+        :param int nb: size of translator
+        :param set((int, int)) duos: set of all group which must be join
+
+        Examples
+        --------
+        >>> NetworkObservations.group_translator(5, ((0, 1), (0, 2), (1, 3)))
+        [3, 3, 3, 3, 5]
+        """
+        translate = arange(nb, dtype="u4")
+        for i, j in sorted(duos):
+            gr_i, gr_j = translate[i], translate[j]
             if gr_i != gr_j:
-                apply_replace(gr_transfer, gr_i, gr_j)
-        return gr_transfer[gr]
+                apply_replace(translate, gr_i, gr_j)
+        return translate
 
     def group_observations(self, **kwargs):
         results, nb_obs = list(), list()

tests/test_network.py

@@ -0,0 +1,15 @@
+from py_eddy_tracker.observations.network import Network
+
+
+def test_group_translate():
+    translate = Network.group_translator(5, ((0, 1), (0, 2), (1, 3)))
+    assert (translate == [3, 3, 3, 3, 4]).all()
+
+    translate = Network.group_translator(5, ((1, 3), (0, 1), (0, 2)))
+    assert (translate == [3, 3, 3, 3, 4]).all()
+
+    translate = Network.group_translator(8, ((1, 3), (2, 3), (2, 4), (5, 6), (4, 5)))
+    assert (translate == [0, 6, 6, 6, 6, 6, 6, 7]).all()
+
+    translate = Network.group_translator(6, ((0, 1), (0, 2), (1, 3), (4, 5)))
+    assert (translate == [3, 3, 3, 3, 5, 5]).all()