Skip to content

Commit b252601

Browse files
committed
TST: Add PeriodIndex Series truncate test
1 parent aa3f9e7 commit b252601

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

pandas/tests/indexes/period/test_indexing.py

+30
Original file line numberDiff line numberDiff line change
@@ -310,3 +310,33 @@ def test_take_fill_value(self):
310310

311311
with pytest.raises(IndexError):
312312
idx.take(np.array([1, -5]))
313+
314+
def test_truncate(self):
315+
# GH 17717
316+
idx1 = pd.PeriodIndex([
317+
pd.Period('2017-09-02'),
318+
pd.Period('2017-09-02'),
319+
pd.Period('2017-09-03')
320+
])
321+
series1 = pd.Series([1, 2, 3], index=idx1)
322+
result1 = series1.truncate(after='2017-09-02')
323+
324+
expected_idx1 = pd.PeriodIndex([
325+
pd.Period('2017-09-02'),
326+
pd.Period('2017-09-02')
327+
])
328+
tm.assert_series_equal(result1, pd.Series([1, 2], index=expected_idx1))
329+
330+
idx2 = pd.PeriodIndex([
331+
pd.Period('2017-09-03'),
332+
pd.Period('2017-09-02'),
333+
pd.Period('2017-09-03')
334+
])
335+
series2 = pd.Series([1, 2, 3], index=idx2)
336+
result2 = series2.truncate(after='2017-09-02')
337+
338+
expected_idx2 = pd.PeriodIndex([
339+
pd.Period('2017-09-03'),
340+
pd.Period('2017-09-02')
341+
])
342+
tm.assert_series_equal(result2, pd.Series([1, 2], index=expected_idx2))

0 commit comments

Comments
 (0)