Skip to content

Commit 60bf12c

Browse files
committed
TST: test get_loc on MultiIndex with tuple containing NaN
1 parent 9709bea commit 60bf12c

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/indexes/test_multi.py

+11
Original file line numberDiff line numberDiff line change
@@ -1243,6 +1243,17 @@ def test_get_loc_level(self):
12431243
assert result == expected
12441244
assert new_index.equals(index.droplevel(0))
12451245

1246+
@pytest.mark.parametrize('level', [0, 1])
1247+
@pytest.mark.parametrize('null_val', [np.nan, pd.NaT, None])
1248+
def test_get_loc_nan(self, level, null_val):
1249+
# GH 18485
1250+
levels = [['a', 'b'], ['c', 'd']]
1251+
key = ['b', 'd']
1252+
levels[level] = np.array([0, null_val], dtype=type(null_val))
1253+
key[level] = null_val
1254+
idx = MultiIndex.from_product(levels)
1255+
assert idx.get_loc(tuple(key)) == 3
1256+
12461257
def test_get_loc_missing_nan(self):
12471258
# GH 8569
12481259
idx = MultiIndex.from_arrays([[1.0, 2.0], [3.0, 4.0]])

0 commit comments

Comments
 (0)