diff --git a/pandas-stubs/core/indexes/base.pyi b/pandas-stubs/core/indexes/base.pyi index 388b3d33e..b8dc94448 100644 --- a/pandas-stubs/core/indexes/base.pyi +++ b/pandas-stubs/core/indexes/base.pyi @@ -292,7 +292,7 @@ class Index(IndexOpsMixin[S1]): @names.setter def names(self, names: list[_str]): ... def set_names(self, names, *, level=..., inplace: bool = ...): ... - def rename(self, name, inplace: bool = ...): ... + def rename(self, name, inplace: bool = ...) -> Self: ... @property def nlevels(self) -> int: ... def sortlevel(self, level=..., ascending: bool = ..., sort_remaining=...): ... diff --git a/tests/test_indexes.py b/tests/test_indexes.py index 84c3c491c..e59fc0ca3 100644 --- a/tests/test_indexes.py +++ b/tests/test_indexes.py @@ -113,6 +113,12 @@ def test_str_split() -> None: check(assert_type(ind.str.split("-", expand=True), pd.MultiIndex), pd.MultiIndex) +def test_index_rename() -> None: + ind = pd.Index([1, 2, 3], name="foo") + ind2 = ind.rename("goo") + check(assert_type(ind2, "pd.Index[int]"), pd.Index, np.integer) + + def test_index_dropna(): idx = pd.Index([1, 2])