66
77import pandas as pd
88from pandas import DataFrame , Series
9+ from pandas .core .groupby .grouper import Grouper
910from pandas .core .indexes .datetimes import date_range
10- from pandas .core .resample import TimeGrouper
1111import pandas .util .testing as tm
1212from pandas .util .testing import assert_frame_equal , assert_series_equal
1313
1616
1717
1818def 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' )
2220
2321 grouped = test_series .groupby (grouper )
2422
@@ -38,9 +36,7 @@ def test_count():
3836
3937 expected = test_series .groupby (lambda x : x .year ).count ()
4038
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' )
4440 result = test_series .groupby (grouper ).count ()
4541 expected .index = result .index
4642 assert_series_equal (result , expected )
@@ -64,7 +60,7 @@ def test_apply_iteration():
6460 N = 1000
6561 ind = pd .date_range (start = "2000-01-01" , freq = "D" , periods = N )
6662 df = DataFrame ({'open' : 1 , 'close' : 2 }, index = ind )
67- tg = TimeGrouper ( 'M' )
63+ tg = Grouper ( freq = 'M' )
6864
6965 _ , grouper , _ = tg ._get_grouper (df )
7066
@@ -93,7 +89,7 @@ def test_fails_on_no_datetime_index(name, func):
9389 msg = ("Only valid with DatetimeIndex, TimedeltaIndex "
9490 "or PeriodIndex, but got an instance of '{}'" .format (name ))
9591 with pytest .raises (TypeError , match = msg ):
96- df .groupby (TimeGrouper ( 'D' ))
92+ df .groupby (Grouper ( freq = 'D' ))
9793
9894
9995def test_aaa_group_order ():
@@ -105,7 +101,7 @@ def test_aaa_group_order():
105101 df ['key' ] = [datetime (2013 , 1 , 1 ), datetime (2013 , 1 , 2 ),
106102 datetime (2013 , 1 , 3 ), datetime (2013 , 1 , 4 ),
107103 datetime (2013 , 1 , 5 )] * 4
108- grouped = df .groupby (TimeGrouper (key = 'key' , freq = 'D' ))
104+ grouped = df .groupby (Grouper (key = 'key' , freq = 'D' ))
109105
110106 tm .assert_frame_equal (grouped .get_group (datetime (2013 , 1 , 1 )),
111107 df [::5 ])
@@ -135,7 +131,7 @@ def test_aggregate_normal(resample_method):
135131 datetime (2013 , 1 , 5 )] * 4
136132
137133 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' ))
139135
140136 expected = getattr (normal_grouped , resample_method )()
141137 dt_result = getattr (dt_grouped , resample_method )()
@@ -195,7 +191,7 @@ def test_aggregate_with_nat(func, fill_value):
195191 datetime (2013 , 1 , 4 ), datetime (2013 , 1 , 5 )] * 4
196192
197193 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' ))
199195
200196 normal_result = getattr (normal_grouped , func )()
201197 dt_result = getattr (dt_grouped , func )()
@@ -222,7 +218,7 @@ def test_aggregate_with_nat_size():
222218 datetime (2013 , 1 , 4 ), datetime (2013 , 1 , 5 )] * 4
223219
224220 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' ))
226222
227223 normal_result = normal_grouped .size ()
228224 dt_result = dt_grouped .size ()
@@ -238,7 +234,7 @@ def test_aggregate_with_nat_size():
238234
239235def test_repr ():
240236 # GH18203
241- result = repr (TimeGrouper (key = 'A' , freq = 'H' ))
237+ result = repr (Grouper (key = 'A' , freq = 'H' ))
242238 expected = ("TimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, "
243239 "closed='left', label='left', how='mean', "
244240 "convention='e', base=0)" )
0 commit comments