From 7ef697cffdcd2f8d701de3cdfd2e6897358effbf Mon Sep 17 00:00:00 2001 From: Tom Augspurger Date: Wed, 3 Oct 2018 13:14:52 -0500 Subject: [PATCH] Use super --- pandas/tests/extension/decimal/test_decimal.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/pandas/tests/extension/decimal/test_decimal.py b/pandas/tests/extension/decimal/test_decimal.py index 036e2d97b4d0f..928ee83df8dc9 100644 --- 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