Fix another test

mroeschke · mroeschke · commit e6e2c89b53aa · 2024-03-06T16:47:45.000-08:00
diff --git a/pandas/tests/computation/test_eval.py b/pandas/tests/computation/test_eval.py
@@ -1784,7 +1784,7 @@ def test_numexpr_option_incompatible_op():
             {"A": [True, False, True, False, None, None], "B": [1, 2, 3, 4, 5, 6]}
         )
         result = df.query("A.isnull()")
-        expected = DataFrame({"A": [None, None], "B": [5, 6]}, index=[4, 5])
+        expected = DataFrame({"A": [None, None], "B": [5, 6]}, index=range(4, 6))
         tm.assert_frame_equal(result, expected)
 
 

Original file line number	Diff line number	Diff line change
`@@ -1784,7 +1784,7 @@ def test_numexpr_option_incompatible_op():`
`1784`	`1784`	`{"A": [True, False, True, False, None, None], "B": [1, 2, 3, 4, 5, 6]}`
`1785`	`1785`	`)`
`1786`	`1786`	`result = df.query("A.isnull()")`
`1787`		`- expected = DataFrame({"A": [None, None], "B": [5, 6]}, index=[4, 5])`
	`1787`	`+ expected = DataFrame({"A": [None, None], "B": [5, 6]}, index=range(4, 6))`
`1788`	`1788`	`tm.assert_frame_equal(result, expected)`
`1789`	`1789`
`1790`	`1790`