@@ -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
167171 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 })
0 commit comments