update run_udf unit tests

Open-EO · Jan 22, 2025 · 00d8a73 · 00d8a73
1 parent 08fc806
commit 00d8a73
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/openeo_processes_dask/process_implementations/udf/udf.py b/openeo_processes_dask/process_implementations/udf/udf.py
@@ -15,7 +15,7 @@ def run_udf(
     data: da.Array, udf: str, runtime: str, context: Optional[dict] = None
 ) -> RasterCube:
     data = XarrayDataCube(xr.DataArray(data))
-    data = UdfData(proj={"EPSG": 900913}, datacube_list=[data], user_context=context)
+    data = UdfData(datacube_list=[data], user_context=context)
     result = run_udf_code(code=udf, data=data)
     cubes = result.get_datacube_list()
     if len(cubes) != 1:

diff --git a/tests/test_udf.py b/tests/test_udf.py
@@ -20,9 +20,9 @@ def test_run_udf(temporal_interval, bounding_box, random_raster_data):
     )
 
     udf = """
-from openeo.udf import XarrayDataCube
-def apply_datacube(cube: XarrayDataCube, context: dict) -> XarrayDataCube:
-    return cube
+import xarray as xr
+def apply_datacube(cube: xr.DataArray, context: dict) -> xr.DataArray:
+    return cube + 1
 """
 
     output_cube = run_udf(data=input_cube, udf=udf, runtime="Python")
@@ -32,7 +32,7 @@ def apply_datacube(cube: XarrayDataCube, context: dict) -> XarrayDataCube:
         output_cube=output_cube,
         verify_attrs=True,
         verify_crs=True,
-        expected_results=input_cube,
+        expected_results=input_cube + 1,
     )
 
-    xr.testing.assert_equal(output_cube, input_cube)
+    xr.testing.assert_equal(output_cube, input_cube + 1)