Bugfix: allow to call ResultQuantityCollection.plot with kwargs. (#94)

Author: gedaskir

CHANGELOG.md

@@ -8,13 +8,14 @@
 
 ### Changed
 
-## [0.6.1] - 2024-03-22
+## [0.6.1] - 2024-03-23
 
 ### Fixed
 
 - Loading MIKE IO 1D together with MIKE+Py
 - Fixed override_name parameter that was not working in ResultFrameAggregator
 - Fixed converting res11 to res1d
+- Fixed calling ResultQuantityCollection.plot with kwargs
 
 ## [0.6] - 2024-02-08
 

mikeio1d/result_network/result_quantity_collection.py

@@ -63,13 +63,15 @@ def read(self, column_mode: Optional[str | ColumnMode] = None) -> pd.DataFrame:
         timeseries_ids = [q.timeseries_id for q in self.result_quantities]
         return self.res1d.read(timeseries_ids, column_mode=column_mode)
 
-    def plot(self):
+    def plot(self, **kwargs):
         """Plot the time series data."""
         if len(self.result_quantities) <= 0:
             return
 
+        # Taking the first data item is enough, because all of them have
+        # the same quantity and for plotting.
         self.data_item = self.result_quantities[0].data_item
-        ResultQuantity.plot(self)
+        return ResultQuantity.plot(self, **kwargs)
 
     def get_query(self):
         """Get queries corresponding to ResultQuantityCollection."""

tests/test_res1d_network_river.py

@@ -390,7 +390,9 @@ def test_result_quantity_collection_methods(test_file):
     assert pytest.approx(max_value) == 100.247
 
     # Test the calling of methods
-    discharge_in_structure.plot()
+    ax = discharge_in_structure.plot()
+    discharge_in_structure.plot(ax=ax)
+
     discharge_in_structure.to_csv(
         file_path.replace("network_river.res1d", "discharge_in_structure.extract.csv")
     )