From d369dacb475db30dd3e6f52f5c5f2419798e0b15 Mon Sep 17 00:00:00 2001 From: dave Date: Thu, 31 Oct 2024 16:54:35 +0100 Subject: [PATCH] install ibis dependency for tests --- tests/load/test_read_interfaces.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/load/test_read_interfaces.py b/tests/load/test_read_interfaces.py index 8e1e588208..0fa1fba5af 100644 --- a/tests/load/test_read_interfaces.py +++ b/tests/load/test_read_interfaces.py @@ -260,6 +260,11 @@ def test_db_cursor_access(populated_pipeline: Pipeline) -> None: ids=lambda x: x.name, ) def test_ibis_tables_access(populated_pipeline: Pipeline) -> None: + # NOTE: we should generalize this with a context for certain deps + import subprocess + + subprocess.check_call(["pip", "install", "ibis-framework[duckdb,postgres,bigquery]"]) + table_relationship = populated_pipeline._dataset().items total_records = _total_records(populated_pipeline) chunk_size = _chunk_size(populated_pipeline) @@ -292,6 +297,11 @@ def test_ibis_tables_access(populated_pipeline: Pipeline) -> None: ids=lambda x: x.name, ) def test_ibis_dataset_access(populated_pipeline: Pipeline) -> None: + # NOTE: we should generalize this with a context for certain deps + import subprocess + + subprocess.check_call(["pip", "install", "ibis-framework[duckdb,postgres,bigquery]"]) + total_records = _total_records(populated_pipeline) ibis_connection = populated_pipeline._ibis()