Skip to content

Commit 1efd00c

Browse files
simonjayhawkinsjreback
authored andcommitted
TST: use index_or_series fixture (#29996)
1 parent f7dea34 commit 1efd00c

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pandas/tests/arithmetic/test_datetime64.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,9 @@ class TestDatetime64SeriesComparison:
171171
],
172172
)
173173
@pytest.mark.parametrize("reverse", [True, False])
174-
@pytest.mark.parametrize("box", [Series, pd.Index])
175174
@pytest.mark.parametrize("dtype", [None, object])
176-
def test_nat_comparisons(self, dtype, box, reverse, pair):
175+
def test_nat_comparisons(self, dtype, index_or_series, reverse, pair):
176+
box = index_or_series
177177
l, r = pair
178178
if reverse:
179179
# add lhs / rhs switched data
@@ -2383,14 +2383,16 @@ def test_dti_add_series(self, tz, names):
23832383
result4 = index + ser.values
23842384
tm.assert_index_equal(result4, expected)
23852385

2386-
@pytest.mark.parametrize("other_box", [pd.Index, Series])
23872386
@pytest.mark.parametrize("op", [operator.add, roperator.radd, operator.sub])
23882387
@pytest.mark.parametrize(
23892388
"names", [(None, None, None), ("foo", "bar", None), ("foo", "foo", "foo")]
23902389
)
2391-
def test_dti_addsub_offset_arraylike(self, tz_naive_fixture, names, op, other_box):
2390+
def test_dti_addsub_offset_arraylike(
2391+
self, tz_naive_fixture, names, op, index_or_series
2392+
):
23922393
# GH#18849, GH#19744
23932394
box = pd.Index
2395+
other_box = index_or_series
23942396
from .test_timedelta64 import get_upcast_box
23952397

23962398
tz = tz_naive_fixture

0 commit comments

Comments
 (0)