Skip to content

Commit 2692932

Browse files
committed
fix test_maybe_convert_i8_numeric II
1 parent cc4859a commit 2692932

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

pandas/tests/indexes/interval/test_interval.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -414,13 +414,18 @@ def test_maybe_convert_i8_nat(self, breaks):
414414
result = index._maybe_convert_i8(to_convert)
415415
tm.assert_index_equal(result, expected)
416416

417-
def test_maybe_convert_i8_numeric(self, any_real_numpy_dtype):
417+
@pytest.mark.parametrize(
418+
"make_key", [lambda breaks: breaks, list],
419+
ids=["lambda", "list"],
420+
)
421+
def test_maybe_convert_i8_numeric(self, make_key, any_real_numpy_dtype):
418422
# GH 20636
419423
breaks = np.arange(5, dtype=any_real_numpy_dtype)
420424
index = IntervalIndex.from_breaks(breaks)
425+
key = make_key(breaks)
421426

422-
result = index._maybe_convert_i8(breaks)
423-
expected = Index(breaks)
427+
result = index._maybe_convert_i8(key)
428+
expected = Index(key)
424429
tm.assert_index_equal(result, expected)
425430

426431
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)