def _get_indexer(self, index: pd.Index) -> Union[Iterable[int], np.ndarray]: indexer = index.get_indexer_for(self.values) missing_mask = indexer == -1 if missing_mask.any(): missing = np.asarray(self.values)[missing_mask].tolist() raise KeyError(missing) return indexer
def test_repeating_interval_index_with_infs(intervals): # GH 46658 interval_index = Index(intervals * 51) expected = np.arange(1, 102, 2, dtype=np.intp) result = interval_index.get_indexer_for([intervals[1]]) tm.assert_equal(result, expected)
def test_get_indexer_non_unique_multiple_nans(idx, target, expected): # GH 35392 axis = Index(idx) actual = axis.get_indexer_for(target) tm.assert_numpy_array_equal(actual, expected)