@@ -160,21 +160,13 @@ def test_query_empty_string(self):
160160 df .query ("" )
161161
162162 def test_query_duplicate_column_name (self , engine , parser ):
163- df = DataFrame (
164- {
165- "A" : range (3 ),
166- "B" : range (3 ),
167- "C" : range (3 )
168- }
169- ).rename (columns = {"B" : "A" })
163+ df = DataFrame ({"A" : range (3 ), "B" : range (3 ), "C" : range (3 )}).rename (
164+ columns = {"B" : "A" }
165+ )
170166
171167 res = df .query ("C == 1" , engine = engine , parser = parser )
172168
173- expect = DataFrame (
174- [[1 , 1 , 1 ]],
175- columns = ["A" , "A" , "C" ],
176- index = [1 ]
177- )
169+ expect = DataFrame ([[1 , 1 , 1 ]], columns = ["A" , "A" , "C" ], index = [1 ])
178170
179171 tm .assert_frame_equal (res , expect )
180172
@@ -1140,9 +1132,7 @@ def test_query_with_nested_special_character(self, parser, engine):
11401132 [">=" , operator .ge ],
11411133 ],
11421134 )
1143- def test_query_lex_compare_strings (
1144- self , parser , engine , op , func
1145- ):
1135+ def test_query_lex_compare_strings (self , parser , engine , op , func ):
11461136 a = Series (np .random .default_rng (2 ).choice (list ("abcde" ), 20 ))
11471137 b = Series (np .arange (a .size ))
11481138 df = DataFrame ({"X" : a , "Y" : b })
0 commit comments