Exposing FFI to python

Author: renato2099

examples/datafusion-ffi-example/python/tests/_test_catalog_provider.py

@@ -22,13 +22,14 @@
 from datafusion import SessionContext
 from datafusion_ffi_example import MyCatalogProvider
 
+
 def test_catalog_provider():
     ctx = SessionContext()
 
     my_catalog_name = "my_catalog"
     expected_schema_name = "my_schema"
     expected_table_name = "my_table"
-    expected_table_columns = ['units', 'price']
+    expected_table_columns = ["units", "price"]
 
     catalog_provider = MyCatalogProvider()
     ctx.register_catalog_provider(my_catalog_name, catalog_provider)
@@ -41,12 +42,9 @@ def test_catalog_provider():
     my_table = my_database.table(expected_table_name)
     assert expected_table_columns == my_table.schema.names
 
-    ctx.register_table(expected_table_name, my_table)
-    expected_df = ctx.sql(f"SELECT * FROM {expected_table_name}").to_pandas()
-    assert len(expected_df) == 5
-    assert expected_table_columns == expected_df.columns.tolist()
-
-    result = ctx.table(f"{my_catalog_name}.{expected_schema_name}.{expected_table_name}").collect()
+    result = ctx.table(
+        f"{my_catalog_name}.{expected_schema_name}.{expected_table_name}"
+    ).collect()
     assert len(result) == 2
 
     col0_result = [r.column(0) for r in result]
@@ -60,4 +58,4 @@ def test_catalog_provider():
         pa.array([1.5, 2.5], type=pa.float64()),
     ]
     assert col0_result == expected_col0
-    assert col1_result == expected_col1
+    assert col1_result == expected_col1

python/datafusion/context.py

@@ -83,7 +83,8 @@ class CatalogProviderExportable(Protocol):
 
     https://docs.rs/datafusion/latest/datafusion/catalog/trait.CatalogProvider.html
     """
-    def __datafusion_catalog_provider__(self) -> object: ...
+
+    def __datafusion_catalog_provider__(self) -> object: ...  # noqa: D105
 
 
 class SessionConfig:
@@ -751,7 +752,7 @@ def deregister_table(self, name: str) -> None:
         self.ctx.deregister_table(name)
 
     def register_catalog_provider(
-            self, name: str, provider: CatalogProviderExportable
+        self, name: str, provider: CatalogProviderExportable
     ) -> None:
         """Register a catalog provider."""
         self.ctx.register_catalog_provider(name, provider)