Improve simplify function to remove succesive nan

Author: AntSimi

src/py_eddy_tracker/generic.py

@@ -315,10 +315,16 @@ def simplify(x, y, precision=0.1):
     if j == (nb - 1):
         return zeros(0, dtype=x.dtype), zeros(0, dtype=x.dtype)
 
+    last_nan = False
     for i in range(j + 1, nb):
         x_, y_ = x[i], y[i]
         if isnan(x_) or isnan(y_):
+            if last_nan:
+                mask[i] = False
+            else:
+                last_nan = True
             continue
+        last_nan = False
         d_x = x_ - x_previous
         if d_x > precision:
             x_previous, y_previous = x_, y_

tests/test_generic.py

@@ -15,6 +15,9 @@ def test_simplify():
     x[4] = nan
     x_, y_ = simplify(x, y, precision=1)
     assert x_.shape[0] == 6
+    x[3] = nan
+    x_, y_ = simplify(x, y, precision=1)
+    assert x_.shape[0] == 6
     x[:4] = nan
     x_, y_ = simplify(x, y, precision=1)
     assert x_.shape[0] == 3