do not push down resample_cube_spatial in case an 'apply' process is in between

Open-EO/openeo-geopyspark-driver#790

Author: jdries

openeo_driver/dry_run.py

@@ -706,6 +706,10 @@ def apply_tiles_spatiotemporal(self, process, context: Optional[dict] = None) ->
         else:
             return self
 
+    def apply(self, process: dict, *, context: Optional[dict] = None, env: EvalEnv) -> "DriverDataCube":
+        cube = self._process("apply", {})
+        return cube
+
     def apply_neighborhood(
         self, process, *, size: List[dict], overlap: List[dict], context: Optional[dict] = None, env: EvalEnv
     ) -> "DriverDataCube":
@@ -771,7 +775,7 @@ def fit_class_random_forest(self, predictors, target, training, num_trees, mtry)
 
     # TODO: some methods need metadata manipulation?
 
-    apply = _nop
+
     apply_tiles = _nop
 
     reduce = _nop

tests/test_dry_run.py

@@ -988,6 +988,25 @@ def test_resample_filter_spatial(dry_run_env, dry_run_tracer):
     }
 
 
+
+def test_resample_cube_spatial(dry_run_env, dry_run_tracer):
+    polygon = {"type": "Polygon", "coordinates": [[(0, 0), (3, 5), (8, 2), (0, 0)]]}
+    cube = DataCube(PGNode("load_collection", id="S2_FOOBAR"), connection=None)
+    cube = cube.apply(lambda x: x + 1)
+    cube = cube.resample_cube_spatial(target=DataCube(PGNode("load_collection", id="S2_FOOBAR"), connection=None),method="max")
+
+    pg = cube.flat_graph()
+    res = evaluate(pg, env=dry_run_env,do_dry_run=False)
+
+    source_constraints = dry_run_tracer.get_source_constraints(merge=True)
+    assert len(source_constraints) == 2
+    src, constraints = source_constraints[0]
+    assert src == ("load_collection", ("S2_FOOBAR", ()))
+
+    assert constraints == { 'process_type': [ ProcessType.FOCAL_SPACE]}
+
+
+
 def test_auto_align(dry_run_env, dry_run_tracer):
     polygon = {"type": "Polygon", "coordinates": [[(0.1, 0.1), (3, 5), (8, 2), (0.1, 0.1)]]}
     cube = DataCube(PGNode("load_collection", id="ESA_WORLDCOVER_10M_2020_V1"), connection=None)