BUG: PeriodEngine doesn't work (#29038)

Author: proost

pandas/core/indexes/period.py

@@ -704,36 +704,30 @@ def get_loc(self, key, method=None, tolerance=None):
         -------
         loc : int
         """
-        try:
-            return self._engine.get_loc(key)
-        except KeyError:
-            if is_integer(key):
-                raise
-
-            try:
-                asdt, parsed, reso = parse_time_string(key, self.freq)
-                key = asdt
-            except TypeError:
-                pass
-            except DateParseError:
-                # A string with invalid format
-                raise KeyError("Cannot interpret '{}' as period".format(key))
+        if is_integer(key) or (is_float(key) and not isna(key)):
+            ordinal = key
+        else:
+            if isinstance(key, str):
+                try:
+                    asdt, parsed, reso = parse_time_string(key, self.freq)
+                    key = asdt
+                except DateParseError:
+                    # A string with invalid format
+                    raise KeyError("Cannot interpret '{}' as period".format(key))
 
             try:
                 key = Period(key, freq=self.freq)
-            except ValueError:
-                # we cannot construct the Period
-                # as we have an invalid type
-                raise KeyError(key)
-
-            try:
                 ordinal = iNaT if key is NaT else key.ordinal
-                if tolerance is not None:
-                    tolerance = self._convert_tolerance(tolerance, np.asarray(key))
-                return self._int64index.get_loc(ordinal, method, tolerance)
+            except ValueError:
+                # we cannot construct the Period as we have an invalid type
+                raise TypeError("'{val}' is an invalid key".format(val=key))
 
-            except KeyError:
-                raise KeyError(key)
+        try:
+            if tolerance is not None:
+                tolerance = self._convert_tolerance(tolerance, np.asarray(key))
+            return self._int64index.get_loc(ordinal, method, tolerance)
+        except KeyError:
+            raise KeyError(key)
 
     def _maybe_cast_slice_bound(self, label, side, kind):
         """

pandas/tests/indexes/common.py

@@ -331,7 +331,7 @@ def test_memory_usage(self, indices):
 
         # RangeIndex, IntervalIndex
         # don't have engines
-        if not isinstance(indices, (RangeIndex, IntervalIndex)):
+        if not isinstance(indices, (RangeIndex, IntervalIndex, PeriodIndex)):
             assert result2 > result
 
         if indices.inferred_type == "object":

pandas/tests/indexes/period/test_indexing.py

@@ -685,3 +685,4 @@ def test_period_index_indexer(self):
         tm.assert_frame_equal(df, df.loc[list(idx)])
         tm.assert_frame_equal(df.iloc[0:5], df.loc[idx[0:5]])
         tm.assert_frame_equal(df, df.loc[list(idx)])
+