Add visvalingam test

Author: AntSimi

src/py_eddy_tracker/observations/network.py

@@ -16,6 +16,7 @@
     in1d,
     ones,
     uint32,
+    uint16,
     unique,
     where,
     zeros,
@@ -513,7 +514,6 @@ def relatives(self, obs, order=2):
             else:
                 segments_connexion[seg][0] = i_slice
 
-
             if i_p != -1:
 
                 if p_seg not in segments_connexion:
@@ -531,15 +531,9 @@ def relatives(self, obs, order=2):
                 segments_connexion[seg][1].append(n_seg)
                 segments_connexion[n_seg][1].append(seg)
 
+        i_obs = [obs] if not hasattr(obs, "__iter__") else obs
 
-        i_obs = (
-            [obs]
-            if not hasattr(obs, "__iter__")
-            else obs
-        )
-        import numpy as np
-
-        distance = zeros(segment.size, dtype=np.uint16) - 1
+        distance = zeros(segment.size, dtype=uint16) - 1
 
         def loop(seg, dist=1):
             i_slice, links = segments_connexion[seg]

src/py_eddy_tracker/observations/observation.py

@@ -2054,8 +2054,8 @@ def contains(self, x, y, intern=False):
         :rtype: array[int32]
         """
         xname, yname = self.intern(intern)
-        m = ~ (isnan(x) + isnan(y))
-        i = -ones(x.shape, dtype='i4')
+        m = ~(isnan(x) + isnan(y))
+        i = -ones(x.shape, dtype="i4")
         i[m] = poly_indexs(x[m], y[m], self[xname], self[yname])
         return i
 

tests/test_poly.py

@@ -1,7 +1,13 @@
 from numpy import array, pi
 from pytest import approx
 
-from py_eddy_tracker.poly import convex, fit_circle, get_convex_hull, poly_area_vertice
+from py_eddy_tracker.poly import (
+    convex,
+    fit_circle,
+    get_convex_hull,
+    poly_area_vertice,
+    visvalingam,
+)
 
 # Vertices for next test
 V = array(((2, 2, 3, 3, 2), (-10, -9, -9, -10, -10)))
@@ -29,3 +35,11 @@ def test_convex():
 
 def test_convex_hull():
     assert convex(*get_convex_hull(*V_concave)) is True
+
+
+def test_visvalingam():
+    x = array([1, 2, 3, 4, 5, 6.75, 6, 1])
+    y = array([-0.5, -1.5, -1, -1.75, -1, -1, -0.5, -0.5])
+    x_, y_ = visvalingam(x, y, 6)
+    assert ([1, 2, 3, 4, 6, 1] == x_).all()
+    assert ([-0.5, -1.5, -1, -1.75, -0.5, -0.5] == y_).all()