Skip to content

Commit 220d25d

Browse files
committed
refactor test function
1 parent 4a6c8ed commit 220d25d

File tree

1 file changed

+24
-8
lines changed

1 file changed

+24
-8
lines changed

pandas/tests/indexes/interval/test_interval.py

Lines changed: 24 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)