Skip to content

Commit 8c170c4

Browse files
authored
CLN: parametrize test_nat_comparisons (#37195)
1 parent bf82227 commit 8c170c4

File tree

1 file changed

+22
-18
lines changed

1 file changed

+22
-18
lines changed

pandas/tests/arithmetic/test_datetime64.py

+22-18
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,26 @@ class TestDatetime64SeriesComparison:
173173
)
174174
@pytest.mark.parametrize("reverse", [True, False])
175175
@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+
):
177196
box = index_or_series
178197
l, r = pair
179198
if reverse:
@@ -182,25 +201,10 @@ def test_nat_comparisons(self, dtype, index_or_series, reverse, pair):
182201

183202
left = Series(l, dtype=dtype)
184203
right = box(r, dtype=dtype)
185-
# Series, Index
186204

187-
expected = Series([False, False, True])
188-
tm.assert_series_equal(left == right, expected)
205+
result = op(left, right)
189206

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)
204208

205209
def test_comparison_invalid(self, tz_naive_fixture, box_with_array):
206210
# GH#4968

0 commit comments

Comments
 (0)