From 4de6d18bde2c19b6cfcb5ffee0c1b503826e1201 Mon Sep 17 00:00:00 2001
From: ValentinaHutter <valentinahutter@gmail.com>
Date: Wed, 18 Sep 2024 16:22:53 +0200
Subject: [PATCH] add array_interpolate

---
 .../process_implementations/arrays.py                  |  2 ++
 tests/test_arrays.py                                   | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/openeo_processes_dask/process_implementations/arrays.py b/openeo_processes_dask/process_implementations/arrays.py
index 3d27f8c..9e89aca 100644
--- 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
index 9af5f90..f6c4565 100644
--- 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():