Add test for rdivmod on EA array (GH23287)

Author: makbigc

pandas/tests/extension/base/ops.py

@@ -1,6 +1,7 @@
 import operator
 
 import pytest
+import numpy as np
 
 import pandas as pd
 from pandas.core import ops
@@ -91,10 +92,18 @@ def test_divmod(self, data):
         self._check_divmod_op(s, divmod, 1, exc=self.divmod_exc)
         self._check_divmod_op(1, ops.rdivmod, s, exc=self.divmod_exc)
 
-    def test_divmod_series_array(self, data):
+    def test_divmod_series_array(self, data, ones):
+        #pytest.set_trace()
         s = pd.Series(data)
         self._check_divmod_op(s, divmod, data)
 
+        #pytest.set_trace()
+        other = ones * 2
+        self._check_divmod_op(other, ops.rdivmod, s)
+
+        other = pd.Series(other)
+        self._check_divmod_op(other, ops.rdivmod, s)
+
     def test_add_series_with_extension_array(self, data):
         s = pd.Series(data)
         result = s + data

pandas/tests/extension/decimal/test_decimal.py

@@ -23,6 +23,11 @@ def data():
     return DecimalArray(make_data())
 
 
+@pytest.fixture
+def ones():
+    return DecimalArray([decimal.Decimal(1) for _ in range(100)])
+
+
 @pytest.fixture
 def data_missing():
     return DecimalArray([decimal.Decimal('NaN'), decimal.Decimal(1)])

pandas/tests/extension/json/test_json.py

@@ -290,6 +290,9 @@ def test_add_series_with_extension_array(self, data):
         with pytest.raises(TypeError, match="unsupported"):
             ser + data
 
+    def test_divmod_series_array(self):
+        pass
+
     def _check_divmod_op(self, s, op, other, exc=NotImplementedError):
         return super(TestArithmeticOps, self)._check_divmod_op(
             s, op, other, exc=TypeError

pandas/tests/extension/test_categorical.py

@@ -214,6 +214,9 @@ def test_add_series_with_extension_array(self, data):
         with pytest.raises(TypeError, match="cannot perform"):
             ser + data
 
+    def test_divmod_series_array(self):
+        pass
+
     def _check_divmod_op(self, s, op, other, exc=NotImplementedError):
         return super(TestArithmeticOps, self)._check_divmod_op(
             s, op, other, exc=TypeError

pandas/tests/extension/test_integer.py

@@ -42,6 +42,11 @@ def data(dtype):
     return integer_array(make_data(), dtype=dtype)
 
 
+@pytest.fixture
+def ones(dtype):
+    return integer_array(np.ones(100), dtype=dtype)
+
+
 @pytest.fixture
 def data_missing(dtype):
     return integer_array([np.nan, 1], dtype=dtype)

pandas/tests/extension/test_period.py

@@ -125,6 +125,9 @@ def test_direct_arith_with_series_returns_not_implemented(self, data):
         result = data.__sub__(other)
         assert result is NotImplemented
 
+    def test_divmod_series_array(self):
+        pass
+
 
 class TestCasting(BasePeriodTests, base.BaseCastingTests):
     pass

pandas/tests/extension/test_sparse.py

@@ -34,6 +34,11 @@ def data(request):
     return res
 
 
+@pytest.fixture
+def ones(request):
+    return SparseArray(np.ones(100))
+
+
 @pytest.fixture(params=[0, np.nan])
 def data_missing(request):
     """Length 2 array with [NA, Valid]"""