@@ -286,6 +286,7 @@ def test_single_bin(self):
286
286
287
287
def test_datetime_cut (self ):
288
288
# GH 14714
289
+ # testing for time data to be present as series
289
290
data = to_datetime (Series (['2013-01-01' , '2013-01-02' , '2013-01-03' ]))
290
291
result , bins = cut (data , 3 , retbins = True )
291
292
expected = Series (['(2012-12-31 23:57:07.200000, 2013-01-01 16:00:00]' ,
@@ -294,37 +295,23 @@ def test_datetime_cut(self):
294
295
).astype ("category" , ordered = True )
295
296
tm .assert_series_equal (result , expected )
296
297
297
- def test_datetime_list_cut (self ):
298
- # GH 14714
298
+ # testing for time data to be present as list
299
299
data = [np .datetime64 ('2013-01-01' ), np .datetime64 ('2013-01-02' ),
300
300
np .datetime64 ('2013-01-03' )]
301
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
302
tm .assert_series_equal (Series (result ), expected )
307
303
308
- def test_datetime_ndarray_cut ( self ):
309
- # GH 14714
304
+ # testing for time data to be present as ndarray
305
+
310
306
data = np .array ([np .datetime64 ('2013-01-01' ),
311
307
np .datetime64 ('2013-01-02' ),
312
308
np .datetime64 ('2013-01-03' )])
313
309
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
310
tm .assert_series_equal (Series (result ), expected )
319
311
320
- def test_datetime_index_cut (self ):
321
- # GH 14714
312
+ # testing for time data to be present as datetime index
322
313
data = DatetimeIndex (['2013-01-01' , '2013-01-02' , '2013-01-03' ])
323
314
result , bins = cut (data , 3 , retbins = True )
324
- expected = Series (['(2012-12-31 23:57:07.200000, 2013-01-01 16:00:00]' ,
325
- '(2013-01-01 16:00:00, 2013-01-02 08:00:00]' ,
326
- '(2013-01-02 08:00:00, 2013-01-03 00:00:00]' ],
327
- ).astype ("category" , ordered = True )
328
315
tm .assert_series_equal (Series (result ), expected )
329
316
330
317
0 commit comments