add frame and series reset_index name checking

Author: minggli

pandas/tests/frame/test_alter_axes.py

@@ -744,6 +744,13 @@ def test_reset_index(self, float_frame):
         xp = xp.set_index(['B'], append=True)
         tm.assert_frame_equal(rs, xp, check_names=False)
 
+    def test_reset_index_name(self):
+        df = DataFrame([[1, 2, 3, 4], [5, 6, 7, 8]],
+                       columns=['A', 'B', 'C', 'D'],
+                       index=Index(range(2), name='x'))
+        assert df.reset_index().index.name == 'x'
+        assert df.reset_index(drop=True).index.name == 'x'
+
     def test_reset_index_level(self):
         df = DataFrame([[1, 2, 3, 4], [5, 6, 7, 8]],
                        columns=['A', 'B', 'C', 'D'])

pandas/tests/series/test_alter_axes.py

@@ -144,6 +144,11 @@ def test_reset_index(self):
         tm.assert_index_equal(rs.index, Index(index.get_level_values(1)))
         assert isinstance(rs, Series)
 
+    def test_reset_index_name(self):
+        s = Series([1, 2, 3], index=Index(range(3), name='x'))
+        assert s.reset_index().index.name == 'x'
+        assert s.reset_index(drop=True).index.name == 'x'
+
     def test_reset_index_level(self):
         df = DataFrame([[1, 2, 3], [4, 5, 6]],
                        columns=['A', 'B', 'C'])