BUG: PeriodIndex.map tries to get super(DatetimIndex, self)

Author: Chang She

pandas/tseries/period.py

@@ -753,7 +753,7 @@ def map(self, func_to_map):
         try:
             return func_to_map(self)
         except:
-            return super(DatetimeIndex, self).map(func_to_map)
+            return super(PeriodIndex, self).map(func_to_map)
 
     def _mpl_repr(self):
         # how to represent ourselves to matplotlib

pandas/tseries/tests/test_period.py

@@ -11,6 +11,7 @@
 import unittest
 
 from numpy.ma.testutils import assert_equal
+from numpy.testing import assert_array_equal
 
 from pandas.tseries.frequencies import MONTHS, DAYS
 from pandas.tseries.period import Period, PeriodIndex, period_range
@@ -1515,6 +1516,12 @@ def test_is_full(self):
 
         self.assert_(index[:0].is_full)
 
+    def test_map(self):
+        index = PeriodIndex([2005, 2007, 2009], freq='A')
+        result = index.map(lambda x: x + 1)
+        expected = index + 1
+        assert_array_equal(result, expected)
+
 def _permute(obj):
     return obj.take(np.random.permutation(len(obj)))