Add 'set_vertical_balance' method to DSheetPilingModel

tests/models/dsheetpiling/test_dsheetpiling_acceptance.py

@@ -34,7 +34,7 @@
     Sheet,
     SheetPileModelPlasticCalculationProperties,
     SheetPileProperties,
-    WoodenSheetPileProperties,
+    WoodenSheetPileProperties, VerticalBalance,
 )
 from geolib.models.dsheetpiling.dsheetpiling_model import (
     DiaphragmModelType,
@@ -198,6 +198,10 @@ def test_run_sheet_model_acceptance_different_calculation_types(
             top_level=level_top, elements=[sheet_element_1, sheet_element_2]
         )
 
+        # Add vertical balance properties
+        vertical_balance = VerticalBalance(max_point_resistance=0.5, xi_factor=1.24)
+        model.set_vertical_balance(vertical_balance=vertical_balance)
+
         # Add soil
         # Set clay material
         soil_clay = Soil(name="Clay", color=Color("green"))

tests/models/dsheetpiling/test_dsheetpiling_model.py

@@ -910,7 +910,7 @@ def test_duplicate_loads(self, model: DSheetPilingModel):
             model.add_surcharge_load(load=testload, side=Side.LEFT, stage_id=0)
 
     @pytest.mark.unittest
-    def test_add_vertical_balance(self, model: DSheetPilingModel):
+    def test_set_vertical_balance(self, model: DSheetPilingModel):
         # Set up vertical balance
         vertical_balance = VerticalBalance(max_point_resistance=1, xi_factor=2)