Fixup

TomAugspurger · Oct 2, 2018 · 0671e7d · 0671e7d
1 parent 1b4261f
commit 0671e7d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 23 deletions.
diff --git a/pandas/tests/extension/decimal/test_decimal.py b/pandas/tests/extension/decimal/test_decimal.py
@@ -8,7 +8,7 @@
 
 from pandas.tests.extension import base
 
-from .array import DecimalDtype, DecimalArray
+from .array import DecimalDtype, DecimalArray, to_decimal
 
 
 def make_data():
@@ -244,6 +244,23 @@ def test_arith_series_with_array(self, data, all_arithmetic_operators):
         context.traps[decimal.DivisionByZero] = divbyzerotrap
         context.traps[decimal.InvalidOperation] = invalidoptrap
 
+    @pytest.mark.parametrize("reverse, expected_div, expected_mod", [
+        (False, [0, 1, 1, 2], [1, 0, 1, 0]),
+        (True, [2, 1, 0, 0], [0, 0, 2, 2]),
+    ])
+    def test_divmod_array(self, reverse, expected_div, expected_mod):
+        # https://github.com/pandas-dev/pandas/issues/22930
+        arr = to_decimal([1, 2, 3, 4])
+        if reverse:
+            div, mod = divmod(2, arr)
+        else:
+            div, mod = divmod(arr, 2)
+        expected_div = to_decimal(expected_div)
+        expected_mod = to_decimal(expected_mod)
+
+        tm.assert_extension_array_equal(div, expected_div)
+        tm.assert_extension_array_equal(mod, expected_mod)
+
     def test_divmod(self, data):
         s = pd.Series(data, name='name')
         a, b = divmod(s, 2)

diff --git a/pandas/tests/extension/test_ops.py b/pandas/tests/extension/test_ops.py