Fix add_delta_tdi return type; docstrings

pandas-dev · jreback · Oct 28, 2018 · Oct 17, 2018 · Oct 17, 2018 · Oct 17, 2018
commit f35b3b68681f4453bdf53456769429f12e1c1af3
diff --git a/pandas/core/arrays/period.py b/pandas/core/arrays/period.py
@@ -745,6 +745,15 @@ def _add_offset(self, other):
         return type(self)(result, freq=self.freq)
 
     def _add_delta_td(self, other):
+        """
+        Parameters
+        ----------
+        other : timedelta, Tick, np.timedelta64
+
+        Returns
+        -------
+        result : ndarray[int64]
+        """
         assert isinstance(self.freq, Tick)  # checked by calling function
         assert isinstance(other, (timedelta, np.timedelta64, Tick))
 
@@ -757,10 +766,19 @@ def _add_delta_td(self, other):
         return ordinals
 
     def _add_delta_tdi(self, other):
+        """
+        Parameters
+        ----------
+        other : TimedeltaArray or ndarray[timedelta64]
+
+        Returns
+        -------
+        result : ndarray[int64]
+        """
         assert isinstance(self.freq, Tick)  # checked by calling function
 
         delta = self._check_timedeltalike_freq_compat(other)
-        return self._addsub_int_array(delta, operator.add)
+        return self._addsub_int_array(delta, operator.add).asi8
 
     def _add_delta(self, other):
         """