Skip to content

Commit ebe5207

Browse files
committed
Fixed Index.rename() and added test
1 parent 7f58060 commit ebe5207

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

pandas-stubs/core/indexes/base.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ class Index(IndexOpsMixin[S1]):
292292
@names.setter
293293
def names(self, names: list[_str]): ...
294294
def set_names(self, names, *, level=..., inplace: bool = ...): ...
295-
def rename(self, name, inplace: bool = ...): ...
295+
def rename(self, name, inplace: bool = ...) -> Self: ...
296296
@property
297297
def nlevels(self) -> int: ...
298298
def sortlevel(self, level=..., ascending: bool = ..., sort_remaining=...): ...

tests/test_indexes.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,11 @@ def test_str_split() -> None:
112112
check(assert_type(ind.str.split("-"), "pd.Index[str]"), pd.Index)
113113
check(assert_type(ind.str.split("-", expand=True), pd.MultiIndex), pd.MultiIndex)
114114

115+
def test_index_rename() -> None:
116+
ind = pd.Index([1, 2, 3], name="foo")
117+
ind2 = ind.rename("goo")
118+
check(assert_type(ind2, pd.Index), pd.Index)
119+
115120

116121
def test_index_dropna():
117122
idx = pd.Index([1, 2])

0 commit comments

Comments
 (0)