6
6
7
7
import pandas as pd
8
8
from pandas import DataFrame , Series
9
+ from pandas .core .groupby .grouper import Grouper
9
10
from pandas .core .indexes .datetimes import date_range
10
- from pandas .core .resample import TimeGrouper
11
11
import pandas .util .testing as tm
12
12
from pandas .util .testing import assert_frame_equal , assert_series_equal
13
13
16
16
17
17
18
18
def test_apply ():
19
- with tm .assert_produces_warning (FutureWarning ,
20
- check_stacklevel = False ):
21
- grouper = pd .TimeGrouper (freq = 'A' , label = 'right' , closed = 'right' )
19
+ grouper = Grouper (freq = 'A' , label = 'right' , closed = 'right' )
22
20
23
21
grouped = test_series .groupby (grouper )
24
22
@@ -38,9 +36,7 @@ def test_count():
38
36
39
37
expected = test_series .groupby (lambda x : x .year ).count ()
40
38
41
- with tm .assert_produces_warning (FutureWarning ,
42
- check_stacklevel = False ):
43
- grouper = pd .TimeGrouper (freq = 'A' , label = 'right' , closed = 'right' )
39
+ grouper = Grouper (freq = 'A' , label = 'right' , closed = 'right' )
44
40
result = test_series .groupby (grouper ).count ()
45
41
expected .index = result .index
46
42
assert_series_equal (result , expected )
@@ -64,7 +60,7 @@ def test_apply_iteration():
64
60
N = 1000
65
61
ind = pd .date_range (start = "2000-01-01" , freq = "D" , periods = N )
66
62
df = DataFrame ({'open' : 1 , 'close' : 2 }, index = ind )
67
- tg = TimeGrouper ( 'M' )
63
+ tg = Grouper ( freq = 'M' )
68
64
69
65
_ , grouper , _ = tg ._get_grouper (df )
70
66
@@ -93,7 +89,7 @@ def test_fails_on_no_datetime_index(name, func):
93
89
msg = ("Only valid with DatetimeIndex, TimedeltaIndex "
94
90
"or PeriodIndex, but got an instance of '{}'" .format (name ))
95
91
with pytest .raises (TypeError , match = msg ):
96
- df .groupby (TimeGrouper ( 'D' ))
92
+ df .groupby (Grouper ( freq = 'D' ))
97
93
98
94
99
95
def test_aaa_group_order ():
@@ -105,7 +101,7 @@ def test_aaa_group_order():
105
101
df ['key' ] = [datetime (2013 , 1 , 1 ), datetime (2013 , 1 , 2 ),
106
102
datetime (2013 , 1 , 3 ), datetime (2013 , 1 , 4 ),
107
103
datetime (2013 , 1 , 5 )] * 4
108
- grouped = df .groupby (TimeGrouper (key = 'key' , freq = 'D' ))
104
+ grouped = df .groupby (Grouper (key = 'key' , freq = 'D' ))
109
105
110
106
tm .assert_frame_equal (grouped .get_group (datetime (2013 , 1 , 1 )),
111
107
df [::5 ])
@@ -135,7 +131,7 @@ def test_aggregate_normal(resample_method):
135
131
datetime (2013 , 1 , 5 )] * 4
136
132
137
133
normal_grouped = normal_df .groupby ('key' )
138
- dt_grouped = dt_df .groupby (TimeGrouper (key = 'key' , freq = 'D' ))
134
+ dt_grouped = dt_df .groupby (Grouper (key = 'key' , freq = 'D' ))
139
135
140
136
expected = getattr (normal_grouped , resample_method )()
141
137
dt_result = getattr (dt_grouped , resample_method )()
@@ -195,7 +191,7 @@ def test_aggregate_with_nat(func, fill_value):
195
191
datetime (2013 , 1 , 4 ), datetime (2013 , 1 , 5 )] * 4
196
192
197
193
normal_grouped = normal_df .groupby ('key' )
198
- dt_grouped = dt_df .groupby (TimeGrouper (key = 'key' , freq = 'D' ))
194
+ dt_grouped = dt_df .groupby (Grouper (key = 'key' , freq = 'D' ))
199
195
200
196
normal_result = getattr (normal_grouped , func )()
201
197
dt_result = getattr (dt_grouped , func )()
@@ -222,7 +218,7 @@ def test_aggregate_with_nat_size():
222
218
datetime (2013 , 1 , 4 ), datetime (2013 , 1 , 5 )] * 4
223
219
224
220
normal_grouped = normal_df .groupby ('key' )
225
- dt_grouped = dt_df .groupby (TimeGrouper (key = 'key' , freq = 'D' ))
221
+ dt_grouped = dt_df .groupby (Grouper (key = 'key' , freq = 'D' ))
226
222
227
223
normal_result = normal_grouped .size ()
228
224
dt_result = dt_grouped .size ()
@@ -238,7 +234,7 @@ def test_aggregate_with_nat_size():
238
234
239
235
def test_repr ():
240
236
# GH18203
241
- result = repr (TimeGrouper (key = 'A' , freq = 'H' ))
237
+ result = repr (Grouper (key = 'A' , freq = 'H' ))
242
238
expected = ("TimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, "
243
239
"closed='left', label='left', how='mean', "
244
240
"convention='e', base=0)" )
0 commit comments