Use super · pandas-dev/pandas@7ef697c (original) (raw)
File tree
1 file changed
lines changed
- pandas/tests/extension/decimal
1 file changed
lines changed
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -261,14 +261,11 @@ def test_divmod_array(self, reverse, expected_div, expected_mod): | ||
| 261 | 261 | tm.assert_extension_array_equal(div, expected_div) |
| 262 | 262 | tm.assert_extension_array_equal(mod, expected_mod) |
| 263 | 263 | |
| 264 | -def test_divmod(self, data): | |
| 265 | -s = pd.Series(data, name='name') | |
| 266 | -a, b = divmod(s, 2) | |
| 267 | -ea, eb = zip(*(divmod(x, 2) for x in s)) | |
| 268 | -ea = pd.Series(ea, name=s.name, dtype=s.dtype) | |
| 269 | -eb = pd.Series(eb, name=s.name, dtype=s.dtype) | |
| 270 | -tm.assert_series_equal(a, ea) | |
| 271 | -tm.assert_series_equal(b, eb) | |
| 264 | +def _check_divmod_op(self, s, op, other, exc=NotImplementedError): | |
| 265 | +# We implement divmod | |
| 266 | +super(TestArithmeticOps, self)._check_divmod_op( | |
| 267 | +s, op, other, exc=None | |
| 268 | + ) | |
| 272 | 269 | |
| 273 | 270 | def test_error(self): |
| 274 | 271 | pass |