Skip geospatial benchmarks if scale not implemented (#1604)

coiled · Nov 15, 2024 · 4c29fdf · 4c29fdf
1 parent 52f7d33
commit 4c29fdf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/geospatial/test_satellite_filtering.py b/tests/geospatial/test_satellite_filtering.py
@@ -1,5 +1,7 @@
 import os
 
+import pytest
+
 from tests.geospatial.workloads.satellite_filtering import satellite_filtering
 
 
@@ -16,6 +18,8 @@ def test_satellite_filtering(
         "large": {"n_workers": 100},
     },
 ):
+    if scale not in scale_kwargs.keys():
+        pytest.skip(reason=f"{scale=} not implemented")
     with setup_benchmark(
         **scale_kwargs[scale],
         env={

diff --git a/tests/geospatial/test_zonal_average.py b/tests/geospatial/test_zonal_average.py
@@ -24,6 +24,8 @@ def test_nwm(
         pytest.skip(
             reason="FIXME: Submission requires pre-computations, but no workers were requested."
         )
+    if scale not in scale_kwargs.keys():
+        pytest.skip(reason=f"{scale=} not implemented")
     with setup_benchmark(
         **scale_kwargs[scale], **cluster_kwargs
     ) as benchmark:  # noqa: F841