updates imports

dlt-hub · Sep 4, 2024 · a6ee746 · a6ee746
1 parent 31fa166
commit a6ee746
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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,