Add ids

coiled · Aug 14, 2024 · 6da5cf7 · 6da5cf7
1 parent 9660f60
commit 6da5cf7
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/tests/benchmarks/test_xarray.py b/tests/benchmarks/test_xarray.py
@@ -39,11 +39,18 @@ def group_reduction_client(
 @pytest.mark.parametrize(
     "func",
     [
-        lambda x: x.groupby("time.month").mean(method="cohorts"),
-        lambda x: x.groupby("time.month").mean(method="map-reduce"),
-        lambda x: x.chunk(time=TimeResampler("ME"))
-        .groupby("time.month")
-        .mean(method="cohorts"),
+        pytest.param(
+            lambda x: x.groupby("time.month").mean(method="cohorts"), id="cohorts"
+        ),
+        pytest.param(
+            lambda x: x.groupby("time.month").mean(method="map-reduce"), id="map-reduce"
+        ),
+        pytest.param(
+            lambda x: x.chunk(time=TimeResampler("ME"))
+            .groupby("time.month")
+            .mean(method="cohorts"),
+            id="chunked-cohorts",
+        ),
     ],
 )
 def test_xarray_groupby_reduction(group_reduction_client, func):