From 79dffa80fd65a2fa125712b7a76959e1928bcba9 Mon Sep 17 00:00:00 2001 From: Austin Raney Date: Mon, 5 Aug 2024 12:09:31 -0400 Subject: [PATCH] test: ngen_cal_model_observations registration --- python/ngen_cal/tests/test_plugin_system.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) 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"""