def test_searchsorted_different_argument_classes(klass): # https://github.com/pandas-dev/pandas/issues/32762 values = IntervalIndex([Interval(0, 1), Interval(1, 2)]) result = values.searchsorted(klass(values)) expected = np.array([0, 1], dtype=result.dtype) tm.assert_numpy_array_equal(result, expected) result = values._data.searchsorted(klass(values)) tm.assert_numpy_array_equal(result, expected)
def test_searchsorted_invalid_argument(arg): values = IntervalIndex([Interval(0, 1), Interval(1, 2)]) msg = "'<' not supported between instances of 'pandas._libs.interval.Interval' and " with pytest.raises(TypeError, match=msg): values.searchsorted(arg)
def test_searchsorted_invalid_argument(arg): values = IntervalIndex([Interval(0, 1), Interval(1, 2)]) msg = "unorderable types" with pytest.raises(TypeError, match=msg): values.searchsorted(arg)