Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_astype.py	4125	0644	editdlrm
test_constructors.py	8663	0644	editdlrm
test_delete.py	2385	0644	editdlrm
test_fillna.py	576	0644	editdlrm
test_formats.py	3275	0644	editdlrm
test_indexing.py	10060	0644	editdlrm
test_insert.py	4022	0644	editdlrm
test_join.py	1497	0644	editdlrm
test_ops.py	11191	0644	editdlrm
test_partial_slicing.py	1347	0644	editdlrm
test_scalar_compat.py	4516	0644	editdlrm
test_searchsorted.py	998	0644	editdlrm
test_setops.py	9371	0644	editdlrm
test_shift.py	2772	0644	editdlrm
test_timedelta.py	7462	0644	editdlrm
test_timedelta_range.py	3258	0644	editdlrm
__init__.py	0	0644	editdlrm

import numpy as np import pytest from pandas import Series, TimedeltaIndex, Timestamp, array import pandas._testing as tm class TestSearchSorted: @pytest.mark.parametrize("klass", [list, np.array, array, Series]) def test_searchsorted_different_argument_classes(self, klass): idx = TimedeltaIndex(["1 day", "2 days", "3 days"]) result = idx.searchsorted(klass(idx)) expected = np.arange(len(idx), dtype=result.dtype) tm.assert_numpy_array_equal(result, expected) result = idx._data.searchsorted(klass(idx)) tm.assert_numpy_array_equal(result, expected) @pytest.mark.parametrize( "arg", [[1, 2], ["a", "b"], [Timestamp("2020-01-01", tz="Europe/London")] * 2], ) def test_searchsorted_invalid_argument_dtype(self, arg): idx = TimedeltaIndex(["1 day", "2 days", "3 days"]) msg = "searchsorted requires compatible dtype" with pytest.raises(TypeError, match=msg): idx.searchsorted(arg)