Wrap add and sub methods with CompWrapper

Author: makbigc

pandas/core/arrays/datetimelike.py

@@ -32,6 +32,7 @@
 from pandas.core.algorithms import (
     checked_add_with_arr, take, unique1d, value_counts)
 import pandas.core.common as com
+from pandas.core.ops import CompWrapper
 
 from pandas.tseries import frequencies
 from pandas.tseries.offsets import DateOffset, Tick
@@ -1175,13 +1176,10 @@ def _time_shift(self, periods, freq=None):
         return self._generate_range(start=start, end=end, periods=None,
                                     freq=self.freq)
 
+    @CompWrapper(zerodim=True, inst_from_senior_cls=True)
     def __add__(self, other):
-        other = lib.item_from_zerodim(other)
-        if isinstance(other, (ABCSeries, ABCDataFrame)):
-            return NotImplemented
-
         # scalar others
-        elif other is NaT:
+        if other is NaT:
             result = self._add_nat()
         elif isinstance(other, (Tick, timedelta, np.timedelta64)):
             result = self._add_delta(other)
@@ -1238,13 +1236,10 @@ def __radd__(self, other):
         # alias for __add__
         return self.__add__(other)
 
+    @CompWrapper(zerodim=True, inst_from_senior_cls=True)
     def __sub__(self, other):
-        other = lib.item_from_zerodim(other)
-        if isinstance(other, (ABCSeries, ABCDataFrame)):
-            return NotImplemented
-
         # scalar others
-        elif other is NaT:
+        if other is NaT:
             result = self._sub_nat()
         elif isinstance(other, (Tick, timedelta, np.timedelta64)):
             result = self._add_delta(-other)