Skip to content

Commit a89f62b

Browse files
committed
Update groupby.py
1 parent c56dd94 commit a89f62b

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

asv_bench/benchmarks/groupby.py

+12-2
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import xarray as xr
55

6-
from . import parameterized, requires_dask
6+
from . import _skip_slow, parameterized, requires_dask
77

88

99
class GroupBy:
@@ -41,13 +41,23 @@ def setup(self, *args, **kwargs):
4141

4242

4343
class GroupByDataFrame(GroupBy):
44+
"""Run groupby tests using pandas DataFrame."""
45+
4446
def setup(self, *args, **kwargs):
47+
# Skip testing in CI as it won't ever change in a commit:
48+
_skip_slow()
49+
4550
super().setup(**kwargs)
4651
self.ds1d = self.ds1d.to_dataframe()
4752

4853

4954
class GroupByDaskDataFrame(GroupBy):
55+
"""Run groupby tests using dask DataFrame."""
56+
5057
def setup(self, *args, **kwargs):
58+
# Skip testing in CI as it won't ever change in a commit:
59+
_skip_slow()
60+
5161
requires_dask()
5262
super().setup(**kwargs)
5363
self.ds1d = self.ds1d.chunk({"dim_0": 50}).to_dataframe()
@@ -75,7 +85,7 @@ def time_agg_small_num_groups(self, method, ndim):
7585
@parameterized(["method", "ndim"], [("sum", "mean"), (1, 2)])
7686
def time_agg_large_num_groups(self, method, ndim):
7787
ds = getattr(self, f"ds{ndim}d")
78-
getattr(ds.resample(time="24H"), method)()
88+
getattr(ds.resample(time="48H"), method)()
7989

8090

8191
class ResampleDask(Resample):

0 commit comments

Comments
 (0)