Skip to content

Commit

Permalink
install ibis dependency for tests
Browse files Browse the repository at this point in the history
  • Loading branch information
sh-rp committed Nov 3, 2024
1 parent 1d78844 commit d369dac
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions tests/load/test_read_interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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()

Expand Down

0 comments on commit d369dac

Please sign in to comment.