add array_interpolate

Open-EO · Sep 18, 2024 · 4de6d18 · 4de6d18
1 parent e7fc372
commit 4de6d18
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/openeo_processes_dask/process_implementations/arrays.py b/openeo_processes_dask/process_implementations/arrays.py
@@ -236,6 +236,8 @@ def array_apply(
 
 
 def array_interpolate_linear(data: ArrayLike):
+    if isinstance(data, list):
+        data = np.array(data)
     x = np.arange(len(data))
     valid = np.isfinite(data)
     if len(x[valid]) < 2:

diff --git a/tests/test_arrays.py b/tests/test_arrays.py
@@ -277,6 +277,16 @@ def test_array_interpolate_linear(data, expected):
         expected,
         equal_nan=True,
     )
+    assert np.array_equal(
+        array_interpolate_linear(np.array(data)),
+        expected,
+        equal_nan=True,
+    )
+    assert np.array_equal(
+        array_interpolate_linear(da.from_array(np.array(data))),
+        expected,
+        equal_nan=True,
+    )
 
 
 def test_first():