diff --git a/pandas/tests/arithmetic/test_datetime64.py b/pandas/tests/arithmetic/test_datetime64.py index cefd2ae7a9ddb..b0b8f1345e4d3 100644 --- a/pandas/tests/arithmetic/test_datetime64.py +++ b/pandas/tests/arithmetic/test_datetime64.py @@ -173,7 +173,26 @@ class TestDatetime64SeriesComparison: ) @pytest.mark.parametrize("reverse", [True, False]) @pytest.mark.parametrize("dtype", [None, object]) - def test_nat_comparisons(self, dtype, index_or_series, reverse, pair): + @pytest.mark.parametrize( + "op, expected", + [ + (operator.eq, Series([False, False, True])), + (operator.ne, Series([True, True, False])), + (operator.lt, Series([False, False, False])), + (operator.gt, Series([False, False, False])), + (operator.ge, Series([False, False, True])), + (operator.le, Series([False, False, True])), + ], + ) + def test_nat_comparisons( + self, + dtype, + index_or_series, + reverse, + pair, + op, + expected, + ): box = index_or_series l, r = pair if reverse: @@ -182,25 +201,10 @@ def test_nat_comparisons(self, dtype, index_or_series, reverse, pair): left = Series(l, dtype=dtype) right = box(r, dtype=dtype) - # Series, Index - expected = Series([False, False, True]) - tm.assert_series_equal(left == right, expected) + result = op(left, right) - expected = Series([True, True, False]) - tm.assert_series_equal(left != right, expected) - - expected = Series([False, False, False]) - tm.assert_series_equal(left < right, expected) - - expected = Series([False, False, False]) - tm.assert_series_equal(left > right, expected) - - expected = Series([False, False, True]) - tm.assert_series_equal(left >= right, expected) - - expected = Series([False, False, True]) - tm.assert_series_equal(left <= right, expected) + tm.assert_series_equal(result, expected) def test_comparison_invalid(self, tz_naive_fixture, box_with_array): # GH#4968