This is an automated email from the ASF dual-hosted git repository.
agrove pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-datafusion-python.git
The following commit(s) were added to refs/heads/master by this push:
new 185ad75 fix lint errors (#18)
185ad75 is described below
commit 185ad757e8474534beb2285173e4c5bf118db86d
Author: Andy Grove <[email protected]>
AuthorDate: Tue Jul 26 12:55:38 2022 +0100
fix lint errors (#18)
---
datafusion/tests/test_context.py | 2 ++
datafusion/tests/test_sql.py | 1 +
2 files changed, 3 insertions(+)
diff --git a/datafusion/tests/test_context.py b/datafusion/tests/test_context.py
index 1e1e771..6c6964c 100644
--- a/datafusion/tests/test_context.py
+++ b/datafusion/tests/test_context.py
@@ -76,6 +76,7 @@ def test_deregister_table(ctx, database):
ctx.deregister_table("csv")
assert public.names() == {"csv1", "csv2"}
+
def test_register_dataset(ctx):
# create a RecordBatch and register it as a pyarrow.dataset.Dataset
batch = pa.RecordBatch.from_arrays(
@@ -92,6 +93,7 @@ def test_register_dataset(ctx):
assert result[0].column(0) == pa.array([5, 7, 9])
assert result[0].column(1) == pa.array([-3, -3, -3])
+
def test_dataset_filter(ctx, capfd):
# create a RecordBatch and register it as a pyarrow.dataset.Dataset
batch = pa.RecordBatch.from_arrays(
diff --git a/datafusion/tests/test_sql.py b/datafusion/tests/test_sql.py
index ffbfc2c..f5e8fbf 100644
--- a/datafusion/tests/test_sql.py
+++ b/datafusion/tests/test_sql.py
@@ -122,6 +122,7 @@ def test_register_parquet_partitioned(ctx, tmp_path):
rd = result.to_pydict()
assert dict(zip(rd["grp"], rd["cnt"])) == {"a": 3, "b": 1}
+
def test_register_dataset(ctx, tmp_path):
path = helpers.write_parquet(tmp_path / "a.parquet", helpers.data())
dataset = ds.dataset(path, format="parquet")
