Skip to content

Commit fbb4e85

Browse files
committed
Add tests.
1 parent 19645b6 commit fbb4e85

File tree

2 files changed

+22
-1
lines changed

2 files changed

+22
-1
lines changed

pandas/core/frame.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5590,7 +5590,7 @@ def reset_index(
55905590
col_level: Hashable = 0,
55915591
col_fill: Hashable = "",
55925592
names: Hashable | Sequence[Hashable] = None,
5593-
) -> Optional[DataFrame]:
5593+
) -> DataFrame | None:
55945594
"""
55955595
Reset the index, or a level of it.
55965596

pandas/tests/frame/methods/test_reset_index.py

+21
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,27 @@ def test_reset_index_name(self):
183183
assert return_value is None
184184
assert df.index.name is None
185185

186+
def test_reset_index_rename(self, float_frame):
187+
# index
188+
rdf = float_frame.reset_index(names="new_name")
189+
exp = Series(float_frame.index.values, name="new_name")
190+
tm.assert_series_equal(rdf["new_name"], exp)
191+
192+
# multiindex
193+
stacked = float_frame.stack()[::2]
194+
stacked = DataFrame({"foo": stacked, "bar": stacked})
195+
196+
names = ["first", "second"]
197+
stacked.index.names = names
198+
deleveled = stacked.reset_index()
199+
deleveled2 = stacked.reset_index(names=["new_first", "new_second"])
200+
tm.assert_series_equal(
201+
deleveled["first"], deleveled2["new_first"], check_names=False
202+
)
203+
tm.assert_series_equal(
204+
deleveled["second"], deleveled2["new_second"], check_names=False
205+
)
206+
186207
def test_reset_index_level(self):
187208
df = DataFrame([[1, 2, 3, 4], [5, 6, 7, 8]], columns=["A", "B", "C", "D"])
188209

0 commit comments

Comments
 (0)