Skip to content

Commit 79caa6f

Browse files
committed
Refactor import of col in test_dataframe.py for improved clarity
1 parent 8a9b171 commit 79caa6f

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

python/tests/test_dataframe.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
WindowFrame,
3434
column,
3535
literal,
36+
col,
3637
)
3738
from datafusion import (
3839
functions as f,
@@ -291,8 +292,6 @@ def test_sort_unsupported(df):
291292

292293

293294
def test_aggregate_string_and_expression_equivalent(df):
294-
from datafusion import col
295-
296295
result_str = df.aggregate("a", [f.count()]).to_pydict()
297296
result_expr = df.aggregate(col("a"), [f.count()]).to_pydict()
298297
assert result_str == result_expr

0 commit comments

Comments
 (0)