@@ -427,23 +427,39 @@ def test_maybe_convert_i8_nat(self, breaks):
427427 [
428428 IntervalIndex .from_breaks ,
429429 lambda breaks : Interval (breaks [0 ], breaks [1 ]),
430- lambda breaks : breaks ,
431430 lambda breaks : breaks [0 ],
432- list ,
433431 ],
434- ids = ["IntervalIndex" , "Interval" , "Index" , " scalar" , "list " ],
432+ ids = ["IntervalIndex" , "Interval" , "scalar" ],
435433 )
436434 def test_maybe_convert_i8_numeric (self , breaks , make_key ):
437435 # GH 20636
438436 index = IntervalIndex .from_breaks (breaks )
439437 key = make_key (breaks )
440438
441439 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 )
447463
448464 @pytest .mark .parametrize (
449465 "breaks1, breaks2" ,
0 commit comments