only import sqlalchemy in tests if present

dlt-hub · Sep 4, 2024 · 2b1578b · 2b1578b
1 parent be0a6c7
commit 2b1578b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/load/sources/sql_database/conftest.py b/tests/load/sources/sql_database/conftest.py
@@ -1,10 +1,15 @@
-from typing import Iterator
+from typing import Iterator, Any
 
 import pytest
 
 import dlt
 from dlt.sources.credentials import ConnectionStringCredentials
-from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB
+
+try:
+    from tests.load.sources.sql_database.sql_source import SQLAlchemySourceDB
+except ModuleNotFoundError:
+    SQLAlchemySourceDB = Any
+    pytest.skip("test needs sql_alchemy")
 
 
 def _create_db(**kwargs) -> Iterator[SQLAlchemySourceDB]:

diff --git a/tests/sources/sql_database/test_sql_database_pipeline_template.py b/tests/sources/sql_database/test_sql_database_pipeline_template.py
@@ -1,5 +1,6 @@
 import pytest
 
+
 # TODO: not all template functions are tested here
 # we may be able to test more in tests/load/sources
 @pytest.mark.parametrize(