From 9833e57ccd2f9ec2090ab1f7da97500a071664b9 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Sat, 7 Mar 2026 09:51:37 -0500
Subject: conftest type annotations.

---
 test_utils/spectrum/conftest.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

(limited to 'test_utils/spectrum/conftest.py')

diff --git a/test_utils/spectrum/conftest.py b/test_utils/spectrum/conftest.py
index b7887f6..d737730 100644
--- a/test_utils/spectrum/conftest.py
+++ b/test_utils/spectrum/conftest.py
@@ -1,24 +1,30 @@
 import logging
-
 import time
+from datetime import datetime, timezone
+from typing import TYPE_CHECKING
 
 import pytest
-from datetime import datetime, timezone
+
 from generalresearch.managers.spectrum.survey import (
-    SpectrumSurveyManager,
     SpectrumCriteriaManager,
+    SpectrumSurveyManager,
 )
 from generalresearch.models.spectrum.survey import SpectrumSurvey
 from generalresearch.sql_helper import SqlHelper
 
-from .surveys_json import SURVEYS_JSON, CONDITIONS
+from .surveys_json import CONDITIONS, SURVEYS_JSON
+
+if TYPE_CHECKING:
+    from generalresearch.config import GRLBaseSettings
 
 
 @pytest.fixture(scope="session")
-def spectrum_rw(settings) -> SqlHelper:
-    print(f"{settings.spectrum_rw_db=}")
+def spectrum_rw(settings: "GRLBaseSettings") -> SqlHelper:
     logging.info(f"{settings.spectrum_rw_db=}")
+
+    assert settings.spectrum_rw_db is not None
     assert "/unittest-" in settings.spectrum_rw_db.path
+
     return SqlHelper(
         dsn=settings.spectrum_rw_db,
         read_timeout=2,
-- 
cgit v1.2.3