diff --git a/python/ngen_cal/tests/test_plugin_system.py b/python/ngen_cal/tests/test_plugin_system.py index f0383670..060f957d 100644 --- a/python/ngen_cal/tests/test_plugin_system.py +++ b/python/ngen_cal/tests/test_plugin_system.py @@ -1,16 +1,18 @@ from __future__ import annotations -from typing import TYPE_CHECKING from types import ModuleType +from typing import TYPE_CHECKING from ngen.cal import hookimpl from ngen.cal._plugin_system import setup_plugin_manager if TYPE_CHECKING: + from datetime import datetime + from pathlib import Path from typing import Callable + import pandas as pd from ngen.cal.configuration import General - from pathlib import Path def test_setup_plugin_manager(): @@ -76,4 +78,14 @@ def ngen_cal_model_output(self) -> None: @hookimpl def ngen_cal_model_post_iteration(self, path: Path, iteration: int) -> None: - """Test model post iteration""" \ No newline at end of file + """Test model post iteration""" + + @hookimpl + def ngen_cal_model_observations( + self, + id: str, + start_time: datetime, + end_time: datetime, + simulation_interval: pd.Timedelta, + ) -> pd.Series: + """Test observation plugin"""