From a6ee746adae048075461d545ae630385def471c3 Mon Sep 17 00:00:00 2001 From: dave Date: Wed, 4 Sep 2024 10:44:56 +0200 Subject: [PATCH] updates imports --- tests/load/sources/sql_database/conftest.py | 2 +- tests/load/sources/sql_database/test_helpers.py | 2 +- tests/load/sources/sql_database/test_sql_database_source.py | 4 ++-- .../test_sql_database_source_all_destinations.py | 6 +++--- 4 files changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/load/sources/sql_database/conftest.py b/tests/load/sources/sql_database/conftest.py index d107216f1c..6b6d1b3946 100644 --- a/tests/load/sources/sql_database/conftest.py +++ b/tests/load/sources/sql_database/conftest.py @@ -4,7 +4,7 @@ import dlt from dlt.sources.credentials import ConnectionStringCredentials -from tests.sources.sql_database.sql_source import SQLAlchemySourceDB +from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB def _create_db(**kwargs) -> Iterator[SQLAlchemySourceDB]: diff --git a/tests/load/sources/sql_database/test_helpers.py b/tests/load/sources/sql_database/test_helpers.py index a32c6c91cd..3522584897 100644 --- a/tests/load/sources/sql_database/test_helpers.py +++ b/tests/load/sources/sql_database/test_helpers.py @@ -6,7 +6,7 @@ from dlt.sources.sql_database.helpers import TableLoader, TableBackend from dlt.sources.sql_database.schema_types import table_to_columns -from tests.sources.sql_database.sql_source import SQLAlchemySourceDB +from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB @pytest.mark.parametrize("backend", ["sqlalchemy", "pyarrow", "pandas", "connectorx"]) diff --git a/tests/load/sources/sql_database/test_sql_database_source.py b/tests/load/sources/sql_database/test_sql_database_source.py index d8758e6371..b594619967 100644 --- a/tests/load/sources/sql_database/test_sql_database_source.py +++ b/tests/load/sources/sql_database/test_sql_database_source.py @@ -26,8 +26,8 @@ assert_schema_on_data, load_tables_to_dicts, ) -from tests.sources.sql_database.sql_source import SQLAlchemySourceDB -from tests.sources.sql_database.test_helpers import mock_json_column +from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB +from tests.load.sources.sql_database.test_helpers import mock_json_column from tests.utils import data_item_length diff --git a/tests/load/sources/sql_database/test_sql_database_source_all_destinations.py b/tests/load/sources/sql_database/test_sql_database_source_all_destinations.py index 303030cf82..0f72db2c82 100644 --- a/tests/load/sources/sql_database/test_sql_database_source_all_destinations.py +++ b/tests/load/sources/sql_database/test_sql_database_source_all_destinations.py @@ -16,9 +16,9 @@ assert_load_info, load_table_counts, ) -from tests.sources.sql_database.sql_source import SQLAlchemySourceDB -from tests.sources.sql_database.test_helpers import mock_json_column -from tests.sources.sql_database.test_sql_database_source import ( +from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB +from tests.load.sources.sql_database.test_helpers import mock_json_column +from tests.load.sources.sql_database.test_sql_database_source import ( assert_row_counts, convert_time_to_us, default_test_callback,