@@ -294,6 +294,29 @@ def test_datetime_cut(self):
294
294
).astype ("category" , ordered = True )
295
295
tm .assert_series_equal (result , expected )
296
296
297
+ def test_datetime_list_cut (self ):
298
+ # GH 14714
299
+ data = [np .datetime64 ('2013-01-01' ), np .datetime64 ('2013-01-02' ),
300
+ np .datetime64 ('2013-01-03' )]
301
+ result , bins = cut (data , 3 , retbins = True )
302
+ expected = Series (['(2012-12-31 23:57:07.200000, 2013-01-01 16:00:00]' ,
303
+ '(2013-01-01 16:00:00, 2013-01-02 08:00:00]' ,
304
+ '(2013-01-02 08:00:00, 2013-01-03 00:00:00]' ],
305
+ ).astype ("category" , ordered = True )
306
+ tm .assert_almost_equal (Series (result ), expected )
307
+
308
+ def test_datetime_ndarray_cut (self ):
309
+ # GH 14714
310
+ data = np .array ([np .datetime64 ('2013-01-01' ),
311
+ np .datetime64 ('2013-01-02' ),
312
+ np .datetime64 ('2013-01-03' )])
313
+ result , bins = cut (data , 3 , retbins = True )
314
+ expected = Series (['(2012-12-31 23:57:07.200000, 2013-01-01 16:00:00]' ,
315
+ '(2013-01-01 16:00:00, 2013-01-02 08:00:00]' ,
316
+ '(2013-01-02 08:00:00, 2013-01-03 00:00:00]' ],
317
+ ).astype ("category" , ordered = True )
318
+ tm .assert_almost_equal (Series (result ), expected )
319
+
297
320
298
321
def curpath ():
299
322
pth , _ = os .path .split (os .path .abspath (__file__ ))
0 commit comments