@@ -171,9 +171,9 @@ class TestDatetime64SeriesComparison:
171
171
],
172
172
)
173
173
@pytest .mark .parametrize ("reverse" , [True , False ])
174
- @pytest .mark .parametrize ("box" , [Series , pd .Index ])
175
174
@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
177
177
l , r = pair
178
178
if reverse :
179
179
# add lhs / rhs switched data
@@ -2383,14 +2383,16 @@ def test_dti_add_series(self, tz, names):
2383
2383
result4 = index + ser .values
2384
2384
tm .assert_index_equal (result4 , expected )
2385
2385
2386
- @pytest .mark .parametrize ("other_box" , [pd .Index , Series ])
2387
2386
@pytest .mark .parametrize ("op" , [operator .add , roperator .radd , operator .sub ])
2388
2387
@pytest .mark .parametrize (
2389
2388
"names" , [(None , None , None ), ("foo" , "bar" , None ), ("foo" , "foo" , "foo" )]
2390
2389
)
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
+ ):
2392
2393
# GH#18849, GH#19744
2393
2394
box = pd .Index
2395
+ other_box = index_or_series
2394
2396
from .test_timedelta64 import get_upcast_box
2395
2397
2396
2398
tz = tz_naive_fixture
0 commit comments