aboutsummaryrefslogtreecommitdiff
path: root/test_utils/spectrum
diff options
context:
space:
mode:
authorMax Nanis2026-03-08 21:57:53 -0400
committerMax Nanis2026-03-08 21:57:53 -0400
commitce291a165fab6b6dc9f053c7b75a699d0fdf389f (patch)
tree7a7b393f432684e13ffe92c40015fb6afc07ceac /test_utils/spectrum
parentb2e56ec3ebc5eb91eb10cc37cc9e02102f441660 (diff)
downloadgeneralresearch-ce291a165fab6b6dc9f053c7b75a699d0fdf389f.tar.gz
generalresearch-ce291a165fab6b6dc9f053c7b75a699d0fdf389f.zip
Simple typing changes, Ruff import formatter. p2
Diffstat (limited to 'test_utils/spectrum')
-rw-r--r--test_utils/spectrum/conftest.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/test_utils/spectrum/conftest.py b/test_utils/spectrum/conftest.py
index d737730..0afc3f5 100644
--- a/test_utils/spectrum/conftest.py
+++ b/test_utils/spectrum/conftest.py
@@ -34,21 +34,21 @@ def spectrum_rw(settings: "GRLBaseSettings") -> SqlHelper:
@pytest.fixture(scope="session")
-def spectrum_criteria_manager(spectrum_rw) -> SpectrumCriteriaManager:
+def spectrum_criteria_manager(spectrum_rw: SqlHelper) -> SpectrumCriteriaManager:
assert "/unittest-" in spectrum_rw.dsn.path
return SpectrumCriteriaManager(spectrum_rw)
@pytest.fixture(scope="session")
-def spectrum_survey_manager(spectrum_rw) -> SpectrumSurveyManager:
+def spectrum_survey_manager(spectrum_rw: SqlHelper) -> SpectrumSurveyManager:
assert "/unittest-" in spectrum_rw.dsn.path
return SpectrumSurveyManager(spectrum_rw)
@pytest.fixture(scope="session")
def setup_spectrum_surveys(
- spectrum_rw, spectrum_survey_manager, spectrum_criteria_manager
-):
+ spectrum_rw: SqlHelper, spectrum_survey_manager, spectrum_criteria_manager
+) -> None:
now = datetime.now(timezone.utc)
# make sure these example surveys exist in db
surveys = [SpectrumSurvey.model_validate_json(x) for x in SURVEYS_JSON]