Moved importorskip() for psycopg in the correct place(s)

agronholm · Aug 2, 2024 · 383f434 · 383f434
1 parent cb1a9b8
commit 383f434
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -156,6 +156,7 @@ async def psycopg_async_store() -> AsyncGenerator[DataStore, None]:
 
     from apscheduler.datastores.sqlalchemy import SQLAlchemyDataStore
 
+    pytest.importorskip("psycopg", reason="psycopg not available")
     engine = create_async_engine(
         "postgresql+psycopg://postgres:secret@localhost/testdb"
     )
@@ -177,6 +178,7 @@ def psycopg_sync_store() -> Generator[DataStore, None, None]:
 
     from apscheduler.datastores.sqlalchemy import SQLAlchemyDataStore
 
+    pytest.importorskip("psycopg", reason="psycopg not available")
     engine = create_engine("postgresql+psycopg://postgres:secret@localhost/testdb")
     try:
         with engine.begin() as conn:

diff --git a/tests/test_datastores.py b/tests/test_datastores.py
@@ -840,7 +840,6 @@ async def test_aiosqlite(
         )
 
     async def test_psycopg(self, psycopg_async_store: SQLAlchemyDataStore) -> None:
-        pytest.importorskip("psycopg")
         assert repr(psycopg_async_store) == (
             "SQLAlchemyDataStore(url='postgresql+psycopg://postgres:***@localhost/"
             "testdb', schema='psycopg_async')"