Skip to content

Commit 335e66d

Browse files
committed
fix: python linter
1 parent 25687e3 commit 335e66d

File tree

1 file changed

+18
-4
lines changed

1 file changed

+18
-4
lines changed

datafusion/tests/test_dataframe.py

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -233,12 +233,19 @@ def test_intersect():
233233
[pa.array([3]), pa.array([6])],
234234
names=["a", "b"],
235235
)
236-
df_c = ctx.create_dataframe([[batch]]).sort(column("a").sort(ascending=True))
236+
df_c = ctx.create_dataframe([[batch]]).sort(
237+
column("a").sort(ascending=True)
238+
)
237239

238240
df_c.show()
239241
df_a.intersect(df_b).sort(column("a").sort(ascending=True)).show()
240242

241-
assert df_c.collect() == df_a.intersect(df_b).sort(column("a").sort(ascending=True)).collect()
243+
assert (
244+
df_c.collect()
245+
== df_a.intersect(df_b)
246+
.sort(column("a").sort(ascending=True))
247+
.collect()
248+
)
242249

243250

244251
def test_except_all():
@@ -260,6 +267,13 @@ def test_except_all():
260267
[pa.array([1, 2]), pa.array([4, 5])],
261268
names=["a", "b"],
262269
)
263-
df_c = ctx.create_dataframe([[batch]]).sort(column("a").sort(ascending=True))
270+
df_c = ctx.create_dataframe([[batch]]).sort(
271+
column("a").sort(ascending=True)
272+
)
264273

265-
assert df_c.collect() == df_a.except_all(df_b).sort(column("a").sort(ascending=True)).collect()
274+
assert (
275+
df_c.collect()
276+
== df_a.except_all(df_b)
277+
.sort(column("a").sort(ascending=True))
278+
.collect()
279+
)

0 commit comments

Comments
 (0)