From 739b1a19a70a92a8c773985762b0e0461079f8fb Mon Sep 17 00:00:00 2001 From: Mohit Mehta Date: Mon, 3 Jul 2023 14:07:57 -0400 Subject: [PATCH] Initial logic --- src/tests/integration/conftest.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/tests/integration/conftest.py b/src/tests/integration/conftest.py index 479e3752f..7f0024a20 100644 --- a/src/tests/integration/conftest.py +++ b/src/tests/integration/conftest.py @@ -4,7 +4,7 @@ from sqlalchemy import MetaData, Table, create_engine from sql import _testing import uuid - +import duckdb def pytest_addoption(parser): parser.addoption("--live", action="store_true") @@ -224,23 +224,23 @@ def ip_with_SQLite(ip_empty, setup_SQLite): @pytest.fixture(scope="session") def setup_duckDB(test_table_name_dict, skip_on_live_mode): - engine = create_engine(_testing.DatabaseConfigHelper.get_database_url("duckDB")) + engine = duckdb.connect(database=":memory:", read_only=False) # Load pre-defined datasets load_generic_testing_data(engine, test_table_name_dict) yield engine tear_down_generic_testing_data(engine, test_table_name_dict) - engine.dispose() + engine.close() @pytest.fixture def ip_with_duckDB(ip_empty, setup_duckDB): configKey = "duckDB" alias = _testing.DatabaseConfigHelper.get_database_config(configKey)["alias"] - + engine = setup_duckDB # Select database engine, use different sqlite database endpoint ip_empty.run_cell( "%sql " - + _testing.DatabaseConfigHelper.get_database_url(configKey) + + f"{engine}" + " --alias " + alias )