Update old test

tm9k1 · Oct 1, 2018 · c92a4a8 · c92a4a8
1 parent 52538fa
commit c92a4a8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pandas/tests/extension/decimal/test_decimal.py b/pandas/tests/extension/decimal/test_decimal.py
@@ -252,9 +252,14 @@ def test_arith_series_with_array(self, data, all_arithmetic_operators):
         context.traps[decimal.DivisionByZero] = divbyzerotrap
         context.traps[decimal.InvalidOperation] = invalidoptrap
 
-    @pytest.mark.skip(reason="divmod not appropriate for decimal")
     def test_divmod(self, data):
-        pass
+        s = pd.Series(data, name='name')
+        a, b = divmod(s, 2)
+        ea, eb = zip(*(divmod(x, 2) for x in s))
+        ea = pd.Series(ea, name=s.name, dtype=s.dtype)
+        eb = pd.Series(eb, name=s.name, dtype=s.dtype)
+        tm.assert_series_equal(a, ea)
+        tm.assert_series_equal(b, eb)
 
     def test_error(self):
         pass