BUG: Added test cases to check loc on a multiindex with nan values pandas-dev#29751

Author: kasim95

pandas/tests/indexing/multiindex/test_getitem.py

@@ -1,6 +1,7 @@
 import numpy as np
 import pytest
 
+import pandas as pd
 from pandas import DataFrame, Index, MultiIndex, Series
 import pandas._testing as tm
 from pandas.core.indexing import IndexingError
@@ -279,3 +280,32 @@ def test_loc_empty_multiindex():
     result = df
     expected = DataFrame([1, 2, 3, 4], index=index, columns=["value"])
     tm.assert_frame_equal(result, expected)
+
+
+@pytest.mark.parametrize("nan", [np.nan, pd.NA, None])
+def test_loc_nan_multiindex(nan):
+    # GH#29751
+    # tests to check loc on a multiindex containing nan values
+    arr = [
+        [11, nan, 13],
+        [21, nan, 23],
+        [31, nan, 33],
+        [41, nan, 43],
+        [51, nan, 53],
+    ]
+    cols = ["a", "b", "c"]
+    df = DataFrame(arr, columns=cols, dtype="int").set_index(["a", "b"])
+    idx = df.index[1]
+    result = df.loc[:idx, :]
+    expected = DataFrame(arr[:2], columns=cols, dtype="int").set_index(["a", "b"])
+    tm.assert_frame_equal(result, expected)
+
+    result = df.loc[idx:, :]
+    expected = DataFrame(arr[1:], columns=cols).set_index(["a", "b"])
+    tm.assert_frame_equal(result, expected)
+
+    idx1 = df.index[1]
+    idx2 = df.index[3]
+    result = df.loc[idx1:idx2, :]
+    expected = DataFrame(arr[1:4], columns=cols).set_index(["a", "b"])
+    tm.assert_frame_equal(result, expected)