bug: Resolve incorrect component use in test & rename variable. (#562)

Deltares · Apr 17, 2024 · 689d592 · 689d592
1 parent e297865
commit 689d592
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hydrolib/core/dimr/models.py b/hydrolib/core/dimr/models.py
@@ -93,10 +93,10 @@ def __init__(self, **data):
         process_input = data.get('process', None)
         self.process = self._set_process_correctly(process_input)
 
-    def _set_process_correctly(self, data):
-        if not data or data == 0:
+    def _set_process_correctly(self, process_input):
+        if not process_input or process_input == 0:
             return None
-        return ' '.join(str(i) for i in range(data))
+        return ' '.join(str(i) for i in range(process_input))
 
     @classmethod
     def get_model(cls):

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -534,7 +534,7 @@ def test_model_diskonlyfilemodel_field_is_constructed_correctly(
     ],
 )
 def test_dimr_with_fmcomponent_saving_process(tmp_path, input_process : int, expected_process_format : str):
-    component = RRComponent(name="test", workingDir='.', inputfile='test.mdu', process=input_process, mpiCommunicator="DFM_COMM_DFMWORLD")
+    component = FMComponent(name="test", workingDir='.', inputfile='test.mdu', process=input_process, mpiCommunicator="DFM_COMM_DFMWORLD")
     dimr = DIMR(component=component)
     save_location : Path = tmp_path/'dimr_config.xml'
     dimr.save(filepath=save_location)