@@ -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