diff options
Diffstat (limited to 'test_utils/spectrum')
| -rw-r--r-- | test_utils/spectrum/conftest.py | 8 |
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] |