updated whatsnew list, added more descriptive tests

Author: GSAUC3

.gitignore

@@ -141,5 +141,3 @@ doc/source/savefig/
 # Pyodide/WASM related files #
 ##############################
 /.pyodide-xbuildenv-*
-
-*.ipynb

doc/source/whatsnew/v3.0.0.rst

@@ -80,14 +80,14 @@ Other enhancements
 - Added half-year offset classes :class:`HalfYearBegin`, :class:`HalfYearEnd`, :class:`BHalfYearBegin` and :class:`BHalfYearEnd` (:issue:`60928`)
 - Added support to read from Apache Iceberg tables with the new :func:`read_iceberg` function (:issue:`61383`)
 - Errors occurring during SQL I/O will now throw a generic :class:`.DatabaseError` instead of the raw Exception type from the underlying driver manager library (:issue:`60748`)
+- Implemented :meth:`MultiIndex.searchsorted` (:issue:`14833`)
 - Implemented :meth:`Series.str.isascii` and :meth:`Series.str.isascii` (:issue:`59091`)
 - Improved deprecation message for offset aliases (:issue:`60820`)
 - Multiplying two :class:`DateOffset` objects will now raise a ``TypeError`` instead of a ``RecursionError`` (:issue:`59442`)
 - Restore support for reading Stata 104-format and enable reading 103-format dta files (:issue:`58554`)
 - Support passing a :class:`Iterable[Hashable]` input to :meth:`DataFrame.drop_duplicates` (:issue:`59237`)
 - Support reading Stata 102-format (Stata 1) dta files (:issue:`58978`)
 - Support reading Stata 110-format (Stata 7) dta files (:issue:`47176`)
-
 .. ---------------------------------------------------------------------------
 .. _whatsnew_300.notable_bug_fixes:
 

pandas/tests/base/test_misc.py

@@ -147,13 +147,6 @@ def test_searchsorted(request, index_or_series_obj):
     # See gh-12238
     obj = index_or_series_obj
 
-    # if isinstance(obj, pd.MultiIndex):
-    #     # See gh-14833
-    #     request.applymarker(
-    #         pytest.mark.xfail(
-    #             reason="np.searchsorted doesn't work on pd.MultiIndex: GH 14833"
-    #         )
-    #     )
     if obj.dtype.kind == "c" and isinstance(obj, Index):
         # TODO: Should Series cases also raise? Looks like they use numpy
         #  comparison semantics https://github.com/numpy/numpy/issues/15981

pandas/tests/indexes/multi/test_indexing.py

@@ -1031,26 +1031,55 @@ def test_get_loc_namedtuple_behaves_like_tuple():
         assert idx.get_loc(("i5", "i6")) == 2
 
 
-def test_searchsorted():
-    # GH14833
-    mi = MultiIndex.from_tuples([("a", 0), ("a", 1), ("b", 0), ("b", 1), ("c", 0)])
+@pytest.fixture
+def mi():
+    return MultiIndex.from_tuples([("a", 0), ("a", 1), ("b", 0), ("b", 1), ("c", 0)])
 
-    assert np.all(mi.searchsorted(("b", 0)) == 2)
-    assert np.all(mi.searchsorted(("b", 0), side="right") == 3)
-    assert np.all(mi.searchsorted(("a", 0)) == 0)
-    assert np.all(mi.searchsorted(("a", -1)) == 0)
-    assert np.all(mi.searchsorted(("c", 1)) == 5)
 
-    result = mi.searchsorted([("a", 1), ("b", 0), ("c", 0)])
-    expected = np.array([1, 2, 4], dtype=np.intp)
-    tm.assert_numpy_array_equal(result, expected)
+@pytest.mark.parametrize(
+    "value, side, expected",
+    [
+        (("b", 0), "left", 2),
+        (("b", 0), "right", 3),
+        (("a", 0), "left", 0),
+        (("a", -1), "left", 0),
+        (("c", 1), "left", 5),
+    ],
+    ids=[
+        "b-0-left",
+        "b-0-right",
+        "a-0-left",
+        "a--1-left",
+        "c-1-left-beyond",
+    ],
+)
+def test_searchsorted_single(mi, value, side, expected):
+    # GH14833
+    result = mi.searchsorted(value, side=side)
+    assert np.all(result == expected)
+
 
-    result = mi.searchsorted([("a", 1), ("b", 0), ("c", 0)], side="right")
-    expected = np.array([2, 3, 5], dtype=np.intp)
+@pytest.mark.parametrize(
+    "values, side, expected",
+    [
+        ([("a", 1), ("b", 0), ("c", 0)], "left", np.array([1, 2, 4], dtype=np.intp)),
+        ([("a", 1), ("b", 0), ("c", 0)], "right", np.array([2, 3, 5], dtype=np.intp)),
+    ],
+    ids=["list-left", "list-right"],
+)
+def test_searchsorted_list(mi, values, side, expected):
+    result = mi.searchsorted(values, side=side)
     tm.assert_numpy_array_equal(result, expected)
 
-    with pytest.raises(ValueError, match="side must be either 'left' or 'right'"):
-        mi.searchsorted(("a", 1), side="middle")
 
-    with pytest.raises(TypeError, match="value must be a tuple or list"):
-        mi.searchsorted("a")
+@pytest.mark.parametrize(
+    "value, side, error_type, match",
+    [
+        (("a", 1), "middle", ValueError, "side must be either 'left' or 'right'"),
+        ("a", "left", TypeError, "value must be a tuple or list"),
+    ],
+    ids=["invalid-side", "invalid-value-type"],
+)
+def test_searchsorted_invalid(mi, value, side, error_type, match):
+    with pytest.raises(error_type, match=match):
+        mi.searchsorted(value, side=side)