port test_td64arr_rfloordiv_tdlike_scalar

pandas/tests/indexes/timedeltas/test_arithmetic.py

@@ -220,26 +220,6 @@ def test_tdi_div_nat_raises(self):
         with pytest.raises(TypeError):
             rng / pd.NaT
 
-    # -------------------------------------------------------------
-    # TimedeltaIndex.__floordiv__
-
-    @pytest.mark.parametrize('scalar_td', [
-        timedelta(minutes=10, seconds=7),
-        Timedelta('10m7s'),
-        Timedelta('10m7s').to_timedelta64()])
-    def test_tdi_floordiv_timedelta_scalar(self, scalar_td):
-        # GH#19125
-        tdi = TimedeltaIndex(['00:05:03', '00:05:03', pd.NaT], freq=None)
-        expected = pd.Index([2.0, 2.0, np.nan])
-
-        res = tdi.__rfloordiv__(scalar_td)
-        tm.assert_index_equal(res, expected)
-
-        expected = pd.Index([0.0, 0.0, np.nan])
-
-        res = tdi // (scalar_td)
-        tm.assert_index_equal(res, expected)
-
 
 class TestTimedeltaIndexArithmetic(object):
     # Addition and Subtraction Operations

pandas/tests/test_arithmetic.py

@@ -47,6 +47,7 @@ def box(request):
     """
     return request.param
 
+
 @pytest.fixture(params=[
     pd.Index,
     Series,
@@ -612,6 +613,30 @@ def test_td64arr_floordiv_tdlike_scalar(self, delta, box_df_fail):
         result = tdi // delta
         tm.assert_equal(result, expected)
 
+    # TODO: Is this redundant with test_td64arr_floordiv_tdlike_scalar?
+    @pytest.mark.parametrize('scalar_td', [
+        timedelta(minutes=10, seconds=7),
+        Timedelta('10m7s'),
+        Timedelta('10m7s').to_timedelta64()
+    ], ids=lambda x: type(x).__name__)
+    def test_td64arr_rfloordiv_tdlike_scalar(self, scalar_td, box_df_fail):
+        # GH#19125
+        box = box_df_fail  # DataFrame op returns m8[ns] instead of f8 dtype
+        tdi = TimedeltaIndex(['00:05:03', '00:05:03', pd.NaT], freq=None)
+        expected = pd.Index([2.0, 2.0, np.nan])
+
+        tdi = tm.box_expected(tdi, box)
+        expected = tm.box_expected(expected, box)
+
+        res = tdi.__rfloordiv__(scalar_td)
+        tm.assert_equal(res, expected)
+
+        expected = pd.Index([0.0, 0.0, np.nan])
+        expected = tm.box_expected(expected, box)
+
+        res = tdi // (scalar_td)
+        tm.assert_equal(res, expected)
+
     # ------------------------------------------------------------------
     # Operations with invalid others