@@ -427,23 +427,39 @@ def test_maybe_convert_i8_nat(self, breaks):
427
427
[
428
428
IntervalIndex .from_breaks ,
429
429
lambda breaks : Interval (breaks [0 ], breaks [1 ]),
430
- lambda breaks : breaks ,
431
430
lambda breaks : breaks [0 ],
432
- list ,
433
431
],
434
- ids = ["IntervalIndex" , "Interval" , "Index" , " scalar" , "list " ],
432
+ ids = ["IntervalIndex" , "Interval" , "scalar" ],
435
433
)
436
434
def test_maybe_convert_i8_numeric (self , breaks , make_key ):
437
435
# GH 20636
438
436
index = IntervalIndex .from_breaks (breaks )
439
437
key = make_key (breaks )
440
438
441
439
result = index ._maybe_convert_i8 (key )
442
- if not isinstance (result , NumericIndex ):
443
- assert result is key
444
- else :
445
- expected = NumericIndex (key )
446
- tm .assert_index_equal (result , expected )
440
+ assert result is key
441
+
442
+ @pytest .mark .parametrize (
443
+ "breaks" ,
444
+ [np .arange (5 , dtype = "int64" ), np .arange (5 , dtype = "float64" )],
445
+ ids = lambda x : str (x .dtype ),
446
+ )
447
+ @pytest .mark .parametrize (
448
+ "make_key" ,
449
+ [
450
+ lambda breaks : breaks ,
451
+ list ,
452
+ ],
453
+ ids = ["Index" , "list" ],
454
+ )
455
+ def test_maybe_convert_i8_numeric_listlike (self , breaks , make_key ):
456
+ # GH 20636
457
+ index = IntervalIndex .from_breaks (breaks )
458
+ key = make_key (breaks )
459
+
460
+ result = index ._maybe_convert_i8 (key )
461
+ expected = NumericIndex (key )
462
+ tm .assert_index_equal (result , expected )
447
463
448
464
@pytest .mark .parametrize (
449
465
"breaks1, breaks2" ,
0 commit comments