Skip to content

Commit d3c52fd

Browse files
committed
Improve tests, add assertion for PeriodIndex as well
1 parent b1f8144 commit d3c52fd

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

pandas/tests/frame/test_timeseries.py

+11-3
Original file line numberDiff line numberDiff line change
@@ -325,10 +325,18 @@ def test_asfreq_datetimeindex(self):
325325

326326
def test_asfreq_datetimeindex_empty_series(self):
327327
# GH 14340
328-
empty = Series(index=pd.DatetimeIndex([])).asfreq('H')
329-
normal = Series(index=pd.DatetimeIndex(["2016-09-29 11:00"]),
328+
329+
# DatetimeIndex
330+
expected = Series(index=pd.DatetimeIndex(["2016-09-29 11:00"])).asfreq('H')
331+
result = Series(index=pd.DatetimeIndex(["2016-09-29 11:00"]),
330332
data=[3]).asfreq('H')
331-
self.assertEqual(empty.index.freq, normal.index.freq)
333+
self.assert_index_equal(expected.index, result.index)
334+
335+
# PeriodIndex
336+
expected = Series(index=pd.PeriodIndex(['2011-1', '2011-2', '2011-3'], freq='M')).asfreq('M')
337+
result = Series(index=pd.PeriodIndex(['2011-1', '2011-2', '2011-3'], freq='M'),
338+
data=[3]).asfreq('M')
339+
self.assert_index_equal(expected.index, result.index)
332340

333341
def test_first_last_valid(self):
334342
N = len(self.frame.index)

0 commit comments

Comments
 (0)