From 6da5cf7c3caa956239e532c01670d34893f29bc2 Mon Sep 17 00:00:00 2001
From: Patrick Hoefler <61934744+phofl@users.noreply.github.com>
Date: Wed, 14 Aug 2024 10:15:36 +0200
Subject: [PATCH] Add ids

---
 tests/benchmarks/test_xarray.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/tests/benchmarks/test_xarray.py b/tests/benchmarks/test_xarray.py
index ba84cb0e41..ad8ff0cc02 100644
--- 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):