@@ -173,7 +173,26 @@ class TestDatetime64SeriesComparison:
173
173
)
174
174
@pytest .mark .parametrize ("reverse" , [True , False ])
175
175
@pytest .mark .parametrize ("dtype" , [None , object ])
176
- def test_nat_comparisons (self , dtype , index_or_series , reverse , pair ):
176
+ @pytest .mark .parametrize (
177
+ "op, expected" ,
178
+ [
179
+ (operator .eq , Series ([False , False , True ])),
180
+ (operator .ne , Series ([True , True , False ])),
181
+ (operator .lt , Series ([False , False , False ])),
182
+ (operator .gt , Series ([False , False , False ])),
183
+ (operator .ge , Series ([False , False , True ])),
184
+ (operator .le , Series ([False , False , True ])),
185
+ ],
186
+ )
187
+ def test_nat_comparisons (
188
+ self ,
189
+ dtype ,
190
+ index_or_series ,
191
+ reverse ,
192
+ pair ,
193
+ op ,
194
+ expected ,
195
+ ):
177
196
box = index_or_series
178
197
l , r = pair
179
198
if reverse :
@@ -182,25 +201,10 @@ def test_nat_comparisons(self, dtype, index_or_series, reverse, pair):
182
201
183
202
left = Series (l , dtype = dtype )
184
203
right = box (r , dtype = dtype )
185
- # Series, Index
186
204
187
- expected = Series ([False , False , True ])
188
- tm .assert_series_equal (left == right , expected )
205
+ result = op (left , right )
189
206
190
- expected = Series ([True , True , False ])
191
- tm .assert_series_equal (left != right , expected )
192
-
193
- expected = Series ([False , False , False ])
194
- tm .assert_series_equal (left < right , expected )
195
-
196
- expected = Series ([False , False , False ])
197
- tm .assert_series_equal (left > right , expected )
198
-
199
- expected = Series ([False , False , True ])
200
- tm .assert_series_equal (left >= right , expected )
201
-
202
- expected = Series ([False , False , True ])
203
- tm .assert_series_equal (left <= right , expected )
207
+ tm .assert_series_equal (result , expected )
204
208
205
209
def test_comparison_invalid (self , tz_naive_fixture , box_with_array ):
206
210
# GH#4968
0 commit comments