@@ -160,13 +160,21 @@ def test_query_empty_string(self):
160160 df .query ("" )
161161
162162 def test_query_duplicate_column_name (self , engine , parser ):
163- df = DataFrame ({"A" : range (3 ), "B" : range (3 ), "C" : range (3 )}).rename (
164- columns = {"B" : "A" }
165- )
163+ df = DataFrame (
164+ {
165+ "A" : range (3 ),
166+ "B" : range (3 ),
167+ "C" : range (3 )
168+ }
169+ ).rename (columns = {"B" : "A" })
166170
167- res = df .query (" C == 1" , engine = engine , parser = parser )
171+ res = df .query (' C == 1' , engine = engine , parser = parser )
168172
169- expect = DataFrame ([[1 , 1 , 1 ]], columns = ["A" , "A" , "C" ], index = [1 ])
173+ expect = DataFrame (
174+ [[1 , 1 , 1 ]],
175+ columns = ["A" , "A" , "C" ],
176+ index = [1 ]
177+ )
170178
171179 tm .assert_frame_equal (res , expect )
172180
@@ -1132,7 +1140,9 @@ def test_query_with_nested_special_character(self, parser, engine):
11321140 [">=" , operator .ge ],
11331141 ],
11341142 )
1135- def test_query_lex_compare_strings (self , parser , engine , op , func ):
1143+ def test_query_lex_compare_strings (
1144+ self , parser , engine , op , func
1145+ ):
11361146 a = Series (np .random .default_rng (2 ).choice (list ("abcde" ), 20 ))
11371147 b = Series (np .arange (a .size ))
11381148 df = DataFrame ({"X" : a , "Y" : b })
@@ -1396,7 +1406,7 @@ def test_expr_with_column_name_with_backtick_and_hash(self):
13961406 def test_expr_with_column_name_with_backtick (self ):
13971407 # GH 59285
13981408 df = DataFrame ({"a`b" : (1 , 2 , 3 ), "ab" : (4 , 5 , 6 )})
1399- result = df .query ("`a``b` < 2" )
1409+ result = df .query ("`a``b` < 2" ) # noqa
14001410 # Note: Formatting checks may wrongly consider the above ``inline code``.
14011411 expected = df [df ["a`b" ] < 2 ]
14021412 tm .assert_frame_equal (result , expected )
0 commit comments