fixed some tests

pyomeca · Nov 13, 2023 · e0c7a0b · e0c7a0b
1 parent 18b58c8
commit e0c7a0b
Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/bioptim/limits/objective_functions.py b/bioptim/limits/objective_functions.py
@@ -387,6 +387,7 @@ def get_type() -> Callable
         MINIMIZE_SEGMENT_ROTATION = (PenaltyFunctionAbstract.Functions.minimize_segment_rotation,)
         MINIMIZE_SEGMENT_VELOCITY = (PenaltyFunctionAbstract.Functions.minimize_segment_velocity,)
         MINIMIZE_STATE = (PenaltyFunctionAbstract.Functions.minimize_states,)
+        MINIMIZE_CONTROL = (PenaltyFunctionAbstract.Functions.minimize_controls,)
         MINIMIZE_TIME = (ObjectiveFunction.MayerFunction.Functions.minimize_time,)
         PROPORTIONAL_STATE = (PenaltyFunctionAbstract.Functions.proportional_states,)
         STOCHASTIC_MINIMIZE_VARIABLE = (PenaltyFunctionAbstract.Functions.stochastic_minimize_variables,)

diff --git a/tests/shard1/test_all_examples.py b/tests/shard1/test_all_examples.py
@@ -1347,7 +1347,7 @@ def test__torque_driven_ocp__trampo_quaternions():
     bioptim_folder = os.path.dirname(ocp_module.__file__)
 
     ocp_module.prepare_ocp(
-        biorbd_model_path=bioptim_folder + "/models/TruncAnd2Arm_Quaternion.bioMod",
+        biorbd_model_path=bioptim_folder + "/models/trunk_and_2arm_quaternion.bioMod",
         n_shooting=5,
         final_time=0.25,
         expand_dynamics=False,

diff --git a/tests/shard3/test_global_torque_driven_ocp.py b/tests/shard3/test_global_torque_driven_ocp.py
@@ -433,7 +433,7 @@ def test_trampo_quaternions(phase_dynamics):
     bioptim_folder = os.path.dirname(ocp_module.__file__)
 
     # Define the problem
-    model_path = bioptim_folder + "/models/TruncAnd2Arm_Quaternion.bioMod"
+    model_path = bioptim_folder + "/models/trunk_and_2arm_quaternion.bioMod"
     final_time = 0.25
     n_shooting = 5