@@ -34,17 +34,17 @@ def time_agg_large_num_groups(self, method, ndim):
34
34
ds = getattr (self , f"ds{ ndim } d" )
35
35
getattr (ds .groupby ("b" ), method )()
36
36
37
- def time_groupby_binary_op_1d (self ):
38
- self .ds1d - self .ds1d_mean
37
+ def time_binary_op_1d (self ):
38
+ self .ds1d . groupby ( "b" ) - self .ds1d_mean
39
39
40
- def time_groupby_binary_op_2d (self ):
41
- self .ds2d - self .ds2d_mean
40
+ def time_binary_op_2d (self ):
41
+ self .ds2d . groupby ( "b" ) - self .ds2d_mean
42
42
43
- def peakmem_groupby_binary_op_1d (self ):
44
- self .ds1d - self .ds1d_mean
43
+ def peakmem_binary_op_1d (self ):
44
+ self .ds1d . groupby ( "b" ) - self .ds1d_mean
45
45
46
- def peakmem_groupby_binary_op_2d (self ):
47
- self .ds2d - self .ds2d_mean
46
+ def peakmem_binary_op_2d (self ):
47
+ self .ds2d . groupby ( "b" ) - self .ds2d_mean
48
48
49
49
50
50
class GroupByDask (GroupBy ):
@@ -71,10 +71,10 @@ def setup(self, *args, **kwargs):
71
71
self .ds1d = self .ds1d .to_dataframe ()
72
72
self .ds1d_mean = self .ds1d .groupby ("b" ).mean ()
73
73
74
- def time_groupby_binary_op_2d (self ):
74
+ def time_binary_op_2d (self ):
75
75
raise NotImplementedError
76
76
77
- def peakmem_groupby_binary_op_2d (self ):
77
+ def peakmem_binary_op_2d (self ):
78
78
raise NotImplementedError
79
79
80
80
@@ -90,10 +90,10 @@ def setup(self, *args, **kwargs):
90
90
self .ds1d = self .ds1d .chunk ({"dim_0" : 50 }).to_dataframe ()
91
91
self .ds1d_mean = self .ds1d .groupby ("b" ).mean ()
92
92
93
- def time_groupby_binary_op_2d (self ):
93
+ def time_binary_op_2d (self ):
94
94
raise NotImplementedError
95
95
96
- def peakmem_groupby_binary_op_2d (self ):
96
+ def peakmem_binary_op_2d (self ):
97
97
raise NotImplementedError
98
98
99
99
@@ -123,18 +123,6 @@ def time_agg_large_num_groups(self, method, ndim):
123
123
ds = getattr (self , f"ds{ ndim } d" )
124
124
getattr (ds .resample (time = "48H" ), method )()
125
125
126
- def time_groupby_binary_op_1d (self ):
127
- self .ds1d - self .ds1d_mean
128
-
129
- def time_groupby_binary_op_2d (self ):
130
- self .ds2d - self .ds2d_mean
131
-
132
- def peakmem_groupby_binary_op_1d (self ):
133
- self .ds1d - self .ds1d_mean
134
-
135
- def peakmem_groupby_binary_op_2d (self ):
136
- self .ds2d - self .ds2d_mean
137
-
138
126
139
127
class ResampleDask (Resample ):
140
128
def setup (self , * args , ** kwargs ):
0 commit comments