test: corrected koswat scenario nans

Deltares · Oct 18, 2023 · cd30d57 · cd30d57
1 parent 0fdeda3
commit cd30d57
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/tests/calculations/test_reinforcement_profile_builder_factory.py b/tests/calculations/test_reinforcement_profile_builder_factory.py
@@ -100,10 +100,10 @@ def _to_koswat_scenario(
                 d_h=_section_scenario.d_h,
                 d_s=_section_scenario.d_s,
                 d_p=_section_scenario.d_p,
-                scenario_section = scenario_data.scenario_dike_section,
-                scenario_name = _section_scenario.scenario_name,
-                kruin_breedte = _section_scenario.kruin_breedte,
-                buiten_talud = _section_scenario.buiten_talud,
+                scenario_section=scenario_data.scenario_dike_section,
+                scenario_name=_section_scenario.scenario_name,
+                kruin_breedte=_section_scenario.kruin_breedte,
+                buiten_talud=_section_scenario.buiten_talud,
             )
             yield _scenario
 
@@ -255,6 +255,15 @@ def reinforcement_profile_case(
             )
         ).build()
 
+        if math.isnan(_combination.koswat_scenario_case.kruin_breedte):
+            _combination.koswat_scenario_case.kruin_breedte = (
+                _combination.input_profile_case.kruin_breedte
+            )
+        if math.isnan(_combination.koswat_scenario_case.buiten_talud):
+            _combination.koswat_scenario_case.buiten_talud = (
+                _combination.input_profile_case.buiten_talud
+            )
+
         def _get_reinforced_profile() -> ReinforcementProfileProtocol:
             _reinforcement = _combination.reinforcement_profile_type()
             # Input profile data.