Skip to content

Commit 37efd51

Browse files
committed
make zero a fixture
1 parent 965f721 commit 37efd51

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

pandas/tests/indexes/test_numeric.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,18 @@
1717

1818
from pandas.tests.indexes.common import Base
1919

20-
# For testing division by (or of) zero for Series with length 5, this
21-
# gives several scalar-zeros and length-5 vector-zeros
20+
2221
zeros = tm.gen_zeros(5)
2322
zeros = [x for x in zeros if not isinstance(x, Series)]
2423

24+
@pytest.fixture(params=zeros)
25+
def zero(request):
26+
"""
27+
For testing division by (or of) zero for Series with length 5, this
28+
gives several scalar-zeros and length-5 vector-zeros
29+
"""
30+
return request.param
31+
2532

2633
def full_like(array, value):
2734
"""Compatibility for numpy<1.8.0
@@ -162,7 +169,6 @@ def test_divmod_series(self):
162169
for r, e in zip(result, expected):
163170
tm.assert_series_equal(r, e)
164171

165-
@pytest.mark.parametrize('zero', zeros)
166172
def test_div_zero(self, zero):
167173
idx = self.create_index()
168174

@@ -173,7 +179,6 @@ def test_div_zero(self, zero):
173179
ser_compat = Series(idx).astype('i8') / np.array(zero).astype('i8')
174180
tm.assert_series_equal(ser_compat, Series(result))
175181

176-
@pytest.mark.parametrize('zero', zeros)
177182
def test_floordiv_zero(self, zero):
178183
idx = self.create_index()
179184
expected = Index([np.nan, np.inf, np.inf, np.inf, np.inf],
@@ -184,7 +189,6 @@ def test_floordiv_zero(self, zero):
184189
ser_compat = Series(idx).astype('i8') // np.array(zero).astype('i8')
185190
tm.assert_series_equal(ser_compat, Series(result))
186191

187-
@pytest.mark.parametrize('zero', zeros)
188192
def test_mod_zero(self, zero):
189193
idx = self.create_index()
190194

@@ -195,7 +199,6 @@ def test_mod_zero(self, zero):
195199
ser_compat = Series(idx).astype('i8') % np.array(zero).astype('i8')
196200
tm.assert_series_equal(ser_compat, Series(result))
197201

198-
@pytest.mark.parametrize('zero', zeros)
199202
def test_divmod_zero(self, zero):
200203
idx = self.create_index()
201204

0 commit comments

Comments
 (0)