@@ -310,3 +310,33 @@ def test_take_fill_value(self):
310
310
311
311
with pytest .raises (IndexError ):
312
312
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