Use super

pandas-dev · Oct 3, 2018 · 7ef697c · 7ef697c
1 parent 35d4213
commit 7ef697c
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/pandas/tests/extension/decimal/test_decimal.py b/pandas/tests/extension/decimal/test_decimal.py
@@ -261,14 +261,11 @@ def test_divmod_array(self, reverse, expected_div, 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)
-        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 _check_divmod_op(self, s, op, other, exc=NotImplementedError):
+        # We implement divmod
+        super(TestArithmeticOps, self)._check_divmod_op(
+            s, op, other, exc=None
+        )
 
     def test_error(self):
         pass