Skip to content

Commit c207b6c

Browse files
committed
ruff formatted
1 parent 9cdd0dc commit c207b6c

File tree

2 files changed

+9
-18
lines changed

2 files changed

+9
-18
lines changed

python/datafusion/dataframe.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,11 +122,9 @@ def __init__(self, df: DataFrameInternal) -> None:
122122
self.df = df
123123

124124
def into_view(self) -> pa.Table:
125-
"""Register this DataFrame as a view in the current session.
126-
127-
"""
125+
"""Register this DataFrame as a view in the current session."""
128126
return self.df.into_view()
129-
127+
130128
def __getitem__(self, key: str | List[str]) -> DataFrame:
131129
"""Return a new :py:class`DataFrame` with the specified column or columns.
132130

python/tests/test_view.py

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,19 @@
22
import pyarrow as pa
33
import pytest
44

5+
56
def test_register_filtered_dataframe():
67
ctx = SessionContext()
78

8-
data = {
9-
"a": [1, 2, 3, 4, 5],
10-
"b": [10, 20, 30, 40, 50]
11-
}
9+
data = {"a": [1, 2, 3, 4, 5], "b": [10, 20, 30, 40, 50]}
1210

1311
df = ctx.from_pydict(data, "my_table")
1412

1513
df_filtered = df.filter(col("a") > literal(2))
1614
view = df_filtered.into_view()
17-
15+
1816
assert view.kind == "view"
19-
17+
2018
ctx.register_table("view1", view)
2119

2220
df_view = ctx.sql("SELECT * FROM view1")
@@ -25,19 +23,14 @@ def test_register_filtered_dataframe():
2523

2624
result_dicts = [batch.to_pydict() for batch in filtered_results]
2725

28-
expected_results = [
29-
{"a": [3, 4, 5], "b": [30, 40, 50]}
30-
]
26+
expected_results = [{"a": [3, 4, 5], "b": [30, 40, 50]}]
3127

3228
assert result_dicts == expected_results
33-
29+
3430
df_results = df.collect()
3531

3632
df_result_dicts = [batch.to_pydict() for batch in df_results]
3733

38-
expected_df_results = [
39-
{"a": [1, 2, 3, 4, 5], "b": [10, 20, 30, 40, 50]}
40-
]
34+
expected_df_results = [{"a": [1, 2, 3, 4, 5], "b": [10, 20, 30, 40, 50]}]
4135

4236
assert df_result_dicts == expected_df_results
43-

0 commit comments

Comments
 (0)