add test for initializing the TimModel with filepath and `quantit…

…ies_names`

hydrolib/core/dflowfm/tim/models.py

@@ -114,6 +114,17 @@ class TimModel(ParsableFileModel):
 
     quantities_names: Optional[List[str]] = Field(default=None)
 
+    def __init__(self, *args, quantities_names: Optional[List[str]] = None, **kwargs):
+        """
+        Custom initializer to handle extra parameters specific to TimModel.
+
+        Args:
+            quantities_names (Optional[List[str]]): Names for the quantities in the timeseries.
+            *args, **kwargs: Other arguments for the superclass.
+        """
+        super().__init__(*args, **kwargs)
+        self.quantities_names = quantities_names
+
     @classmethod
     def _ext(cls) -> str:
         return ".tim"

tests/dflowfm/test_tim.py

@@ -106,6 +106,12 @@ def test_initialization(self):
         assert len(model.timeseries) == 0
         assert model.as_dataframe().empty
 
+    def test_initialize_with_quantities_names(self, input_files_dir: Path):
+        path = input_files_dir / "tim/single_data_for_timeseries.tim"
+        model = TimModel(path, quantities_names=["a"])
+        assert model.quantities_names == ["a"]
+        assert len(model.timeseries) == 13
+
     def test_as_dataframe(self):
         model = TimModel(
             timeseries=self.single_data_for_timeseries_floats,