11
11
from pandas .core .indexes .period import PeriodIndex , period_range
12
12
from pandas .core .indexes .timedeltas import TimedeltaIndex , timedelta_range
13
13
import pandas .util .testing as tm
14
- from pandas .util .testing import (
15
- assert_almost_equal ,
16
- assert_frame_equal ,
17
- assert_index_equal ,
18
- assert_series_equal ,
19
- )
20
14
21
15
# a fixture value can be overridden by the test parameter value. Note that the
22
16
# value of the fixture can be overridden this way even if the test doesn't use
@@ -53,7 +47,7 @@ def test_asfreq(series_and_frame, freq, create_index):
53
47
result = obj .resample (freq ).asfreq ()
54
48
new_index = create_index (obj .index [0 ], obj .index [- 1 ], freq = freq )
55
49
expected = obj .reindex (new_index )
56
- assert_almost_equal (result , expected )
50
+ tm . assert_almost_equal (result , expected )
57
51
58
52
59
53
@pytest .mark .parametrize (
@@ -67,21 +61,21 @@ def test_asfreq_fill_value(series, create_index):
67
61
result = s .resample ("1H" ).asfreq ()
68
62
new_index = create_index (s .index [0 ], s .index [- 1 ], freq = "1H" )
69
63
expected = s .reindex (new_index )
70
- assert_series_equal (result , expected )
64
+ tm . assert_series_equal (result , expected )
71
65
72
66
frame = s .to_frame ("value" )
73
67
frame .iloc [1 ] = None
74
68
result = frame .resample ("1H" ).asfreq (fill_value = 4.0 )
75
69
new_index = create_index (frame .index [0 ], frame .index [- 1 ], freq = "1H" )
76
70
expected = frame .reindex (new_index , fill_value = 4.0 )
77
- assert_frame_equal (result , expected )
71
+ tm . assert_frame_equal (result , expected )
78
72
79
73
80
74
@all_ts
81
75
def test_resample_interpolate (frame ):
82
76
# # 12925
83
77
df = frame
84
- assert_frame_equal (
78
+ tm . assert_frame_equal (
85
79
df .resample ("1T" ).asfreq ().interpolate (), df .resample ("1T" ).interpolate ()
86
80
)
87
81
@@ -113,9 +107,9 @@ def test_resample_empty_series(freq, empty_series, resample_method):
113
107
expected .index = s .index .asfreq (freq = freq )
114
108
else :
115
109
expected .index = s .index ._shallow_copy (freq = freq )
116
- assert_index_equal (result .index , expected .index )
110
+ tm . assert_index_equal (result .index , expected .index )
117
111
assert result .index .freq == expected .index .freq
118
- assert_series_equal (result , expected , check_dtype = False )
112
+ tm . assert_series_equal (result , expected , check_dtype = False )
119
113
120
114
121
115
@all_ts
@@ -135,9 +129,9 @@ def test_resample_empty_dataframe(empty_frame, freq, resample_method):
135
129
expected .index = df .index .asfreq (freq = freq )
136
130
else :
137
131
expected .index = df .index ._shallow_copy (freq = freq )
138
- assert_index_equal (result .index , expected .index )
132
+ tm . assert_index_equal (result .index , expected .index )
139
133
assert result .index .freq == expected .index .freq
140
- assert_almost_equal (result , expected , check_dtype = False )
134
+ tm . assert_almost_equal (result , expected , check_dtype = False )
141
135
142
136
# test size for GH13212 (currently stays as df)
143
137
@@ -186,12 +180,12 @@ def test_resample_loffset_arg_type(frame, create_index):
186
180
if isinstance (expected .index , TimedeltaIndex ):
187
181
msg = "DataFrame are different"
188
182
with pytest .raises (AssertionError , match = msg ):
189
- assert_frame_equal (result_agg , expected )
183
+ tm . assert_frame_equal (result_agg , expected )
190
184
with pytest .raises (AssertionError , match = msg ):
191
- assert_frame_equal (result_how , expected )
185
+ tm . assert_frame_equal (result_how , expected )
192
186
else :
193
- assert_frame_equal (result_agg , expected )
194
- assert_frame_equal (result_how , expected )
187
+ tm . assert_frame_equal (result_agg , expected )
188
+ tm . assert_frame_equal (result_how , expected )
195
189
196
190
197
191
@all_ts
@@ -202,7 +196,7 @@ def test_apply_to_empty_series(empty_series):
202
196
result = s .resample (freq ).apply (lambda x : 1 )
203
197
expected = s .resample (freq ).apply (np .sum )
204
198
205
- assert_series_equal (result , expected , check_dtype = False )
199
+ tm . assert_series_equal (result , expected , check_dtype = False )
206
200
207
201
208
202
@all_ts
@@ -214,7 +208,7 @@ def test_resampler_is_iterable(series):
214
208
resampled = series .resample (freq )
215
209
for (rk , rv ), (gk , gv ) in zip (resampled , grouped ):
216
210
assert rk == gk
217
- assert_series_equal (rv , gv )
211
+ tm . assert_series_equal (rv , gv )
218
212
219
213
220
214
@all_ts
0 commit comments