Skip to content

Commit 8e0068a

Browse files
author
cloudboat
committed
FIX: Revert accidental changes to test_query_eval.py
1 parent c4ecf7a commit 8e0068a

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

pandas/tests/frame/test_query_eval.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)