moving examples folder inside bioptim

pyomeca · Nov 10, 2021 · b4c8e84 · b4c8e84
1 parent 91fb017
commit b4c8e84
Show file tree

Hide file tree

Showing 113 changed files with 726 additions and 531 deletions.
diff --git a/.gitignore b/.gitignore
@@ -94,7 +94,7 @@ examples/*/biorbd_optim
 examples/biorbd_optim
 
 # Remove other development artifacts
-examples/sandbox
+*/examples/sandbox
 *.npy
 *.bo
 *.bob

diff --git a/MANIFEST.in b/MANIFEST.in
@@ -6,5 +6,7 @@ recursive-include examples *.py
 recursive-include examples *.bioMod
 recursive-include examples *.vtp
 recursive-include examples *.STL
-recursive-exclude examples/sandbox *.*
-recursive-exclude examples c_generated_code
+prune *sandbox*
+prune *c_generated_code*
+global-exclude *.py[cod]
+
diff --git a/__init__.py b/__init__.py
diff --git a/bioptim/examples/__init__.py b/bioptim/examples/__init__.py
diff --git a/examples/__main__.py → bioptim/examples/__main__.py b/examples/__main__.py → bioptim/examples/__main__.py
diff --git a/examples/acados/cube.py → bioptim/examples/acados/cube.py b/examples/acados/cube.py → bioptim/examples/acados/cube.py
diff --git a/examples/acados/models/arm26.bioMod → bioptim/examples/acados/models/arm26.bioMod b/examples/acados/models/arm26.bioMod → bioptim/examples/acados/models/arm26.bioMod
diff --git a/examples/acados/models/cube.bioMod → bioptim/examples/acados/models/cube.bioMod b/examples/acados/models/cube.bioMod → bioptim/examples/acados/models/cube.bioMod
diff --git a/...ples/acados/models/mesh/arm_r_humerus.vtp → ...ples/acados/models/mesh/arm_r_humerus.vtp b/...ples/acados/models/mesh/arm_r_humerus.vtp → ...ples/acados/models/mesh/arm_r_humerus.vtp
diff --git a/examples/acados/models/mesh/arm_r_ulna.vtp → ...xamples/acados/models/mesh/arm_r_ulna.vtp b/examples/acados/models/mesh/arm_r_ulna.vtp → ...xamples/acados/models/mesh/arm_r_ulna.vtp
diff --git a/examples/acados/models/mesh/ground_ribs.vtp → ...amples/acados/models/mesh/ground_ribs.vtp b/examples/acados/models/mesh/ground_ribs.vtp → ...amples/acados/models/mesh/ground_ribs.vtp
diff --git a/examples/acados/models/mesh/pendulum.vtp → .../examples/acados/models/mesh/pendulum.vtp b/examples/acados/models/mesh/pendulum.vtp → .../examples/acados/models/mesh/pendulum.vtp
diff --git a/examples/acados/models/pendulum.bioMod → ...im/examples/acados/models/pendulum.bioMod b/examples/acados/models/pendulum.bioMod → ...im/examples/acados/models/pendulum.bioMod
diff --git a/examples/acados/pendulum.py → bioptim/examples/acados/pendulum.py b/examples/acados/pendulum.py → bioptim/examples/acados/pendulum.py
diff --git a/examples/acados/static_arm.py → bioptim/examples/acados/static_arm.py b/examples/acados/static_arm.py → bioptim/examples/acados/static_arm.py
diff --git a/...ples/fatigue/models/arm26_constant.bioMod → ...ples/fatigue/models/arm26_constant.bioMod b/...ples/fatigue/models/arm26_constant.bioMod → ...ples/fatigue/models/arm26_constant.bioMod
diff --git a/...les/fatigue/models/mesh/arm_r_humerus.vtp → ...les/fatigue/models/mesh/arm_r_humerus.vtp b/...les/fatigue/models/mesh/arm_r_humerus.vtp → ...les/fatigue/models/mesh/arm_r_humerus.vtp
diff --git a/examples/fatigue/models/mesh/arm_r_ulna.vtp → ...amples/fatigue/models/mesh/arm_r_ulna.vtp b/examples/fatigue/models/mesh/arm_r_ulna.vtp → ...amples/fatigue/models/mesh/arm_r_ulna.vtp
diff --git a/examples/fatigue/models/mesh/ground_ribs.vtp → ...mples/fatigue/models/mesh/ground_ribs.vtp b/examples/fatigue/models/mesh/ground_ribs.vtp → ...mples/fatigue/models/mesh/ground_ribs.vtp
diff --git a/examples/fatigue/models/mesh/pendulum.vtp → ...examples/fatigue/models/mesh/pendulum.vtp b/examples/fatigue/models/mesh/pendulum.vtp → ...examples/fatigue/models/mesh/pendulum.vtp
diff --git a/examples/fatigue/models/pendulum.bioMod → ...m/examples/fatigue/models/pendulum.bioMod b/examples/fatigue/models/pendulum.bioMod → ...m/examples/fatigue/models/pendulum.bioMod
diff --git a/examples/fatigue/pendulum_with_fatigue.py → ...examples/fatigue/pendulum_with_fatigue.py b/examples/fatigue/pendulum_with_fatigue.py → ...examples/fatigue/pendulum_with_fatigue.py
diff --git a/examples/fatigue/static_arm_with_fatigue.py → ...amples/fatigue/static_arm_with_fatigue.py b/examples/fatigue/static_arm_with_fatigue.py → ...amples/fatigue/static_arm_with_fatigue.py
diff --git a/examples/getting_started/custom_bounds.py → ...examples/getting_started/custom_bounds.py b/examples/getting_started/custom_bounds.py → ...examples/getting_started/custom_bounds.py
diff --git a/...ples/getting_started/custom_constraint.py → ...ples/getting_started/custom_constraint.py b/...ples/getting_started/custom_constraint.py → ...ples/getting_started/custom_constraint.py
diff --git a/examples/getting_started/custom_dynamics.py → ...amples/getting_started/custom_dynamics.py b/examples/getting_started/custom_dynamics.py → ...amples/getting_started/custom_dynamics.py
diff --git a/...s/getting_started/custom_initial_guess.py → ...s/getting_started/custom_initial_guess.py b/...s/getting_started/custom_initial_guess.py → ...s/getting_started/custom_initial_guess.py
diff --git a/...ples/getting_started/custom_objectives.py → ...ples/getting_started/custom_objectives.py b/...ples/getting_started/custom_objectives.py → ...ples/getting_started/custom_objectives.py
diff --git a/...ples/getting_started/custom_parameters.py → ...ples/getting_started/custom_parameters.py b/...ples/getting_started/custom_parameters.py → ...ples/getting_started/custom_parameters.py
diff --git a/...tting_started/custom_phase_transitions.py → ...tting_started/custom_phase_transitions.py b/...tting_started/custom_phase_transitions.py → ...tting_started/custom_phase_transitions.py
diff --git a/examples/getting_started/custom_plotting.py → ...amples/getting_started/custom_plotting.py b/examples/getting_started/custom_plotting.py → ...amples/getting_started/custom_plotting.py
diff --git a/...etting_started/example_cyclic_movement.py → ...etting_started/example_cyclic_movement.py b/...etting_started/example_cyclic_movement.py → ...etting_started/example_cyclic_movement.py
diff --git a/...etting_started/example_external_forces.py → ...etting_started/example_external_forces.py b/...etting_started/example_external_forces.py → ...etting_started/example_external_forces.py
diff --git a/..._started/example_inequality_constraint.py → ..._started/example_inequality_constraint.py b/..._started/example_inequality_constraint.py → ..._started/example_inequality_constraint.py
diff --git a/examples/getting_started/example_mapping.py → ...amples/getting_started/example_mapping.py b/examples/getting_started/example_mapping.py → ...amples/getting_started/example_mapping.py
diff --git a/...les/getting_started/example_multiphase.py → ...les/getting_started/example_multiphase.py b/...les/getting_started/example_multiphase.py → ...les/getting_started/example_multiphase.py
diff --git a/...s/getting_started/example_optimal_time.py → ...s/getting_started/example_optimal_time.py b/...s/getting_started/example_optimal_time.py → ...s/getting_started/example_optimal_time.py
diff --git a/.../getting_started/example_save_and_load.py → .../getting_started/example_save_and_load.py b/.../getting_started/example_save_and_load.py → .../getting_started/example_save_and_load.py
diff --git a/...les/getting_started/example_simulation.py → ...les/getting_started/example_simulation.py b/...les/getting_started/example_simulation.py → ...les/getting_started/example_simulation.py
diff --git a/...ed/models/2segments_4dof_2contacts.bioMod → ...ed/models/2segments_4dof_2contacts.bioMod b/...ed/models/2segments_4dof_2contacts.bioMod → ...ed/models/2segments_4dof_2contacts.bioMod
diff --git a/examples/getting_started/models/cube.bioMod → ...amples/getting_started/models/cube.bioMod b/examples/getting_started/models/cube.bioMod → ...amples/getting_started/models/cube.bioMod
diff --git a/...ng_started/models/cube_with_forces.bioMod → ...ng_started/models/cube_with_forces.bioMod b/...ng_started/models/cube_with_forces.bioMod → ...ng_started/models/cube_with_forces.bioMod
diff --git a/.../getting_started/models/mesh/pendulum.STL → .../getting_started/models/mesh/pendulum.STL b/.../getting_started/models/mesh/pendulum.STL → .../getting_started/models/mesh/pendulum.STL
diff --git a/...es/getting_started/models/pendulum.bioMod → ...es/getting_started/models/pendulum.bioMod b/...es/getting_started/models/pendulum.bioMod → ...es/getting_started/models/pendulum.bioMod
diff --git a/examples/getting_started/pendulum.py → bioptim/examples/getting_started/pendulum.py b/examples/getting_started/pendulum.py → bioptim/examples/getting_started/pendulum.py
diff --git a/.../moving_horizon_estimation/cyclic_nmpc.py → .../moving_horizon_estimation/cyclic_nmpc.py b/.../moving_horizon_estimation/cyclic_nmpc.py → .../moving_horizon_estimation/cyclic_nmpc.py
diff --git a/examples/moving_horizon_estimation/mhe.py → ...examples/moving_horizon_estimation/mhe.py b/examples/moving_horizon_estimation/mhe.py → ...examples/moving_horizon_estimation/mhe.py
diff --git a/...ing_horizon_estimation/models/arm2.bioMod → ...ing_horizon_estimation/models/arm2.bioMod b/...ing_horizon_estimation/models/arm2.bioMod → ...ing_horizon_estimation/models/arm2.bioMod
diff --git a/...on_estimation/models/cart_pendulum.bioMod → ...on_estimation/models/cart_pendulum.bioMod b/...on_estimation/models/cart_pendulum.bioMod → ...on_estimation/models/cart_pendulum.bioMod
diff --git a/..._estimation/models/mesh/arm_r_humerus.vtp → ..._estimation/models/mesh/arm_r_humerus.vtp b/..._estimation/models/mesh/arm_r_humerus.vtp → ..._estimation/models/mesh/arm_r_humerus.vtp
diff --git a/...zon_estimation/models/mesh/arm_r_ulna.vtp → ...zon_estimation/models/mesh/arm_r_ulna.vtp b/...zon_estimation/models/mesh/arm_r_ulna.vtp → ...zon_estimation/models/mesh/arm_r_ulna.vtp
diff --git a/...on_estimation/models/mesh/ground_ribs.vtp → ...on_estimation/models/mesh/ground_ribs.vtp b/...on_estimation/models/mesh/ground_ribs.vtp → ...on_estimation/models/mesh/ground_ribs.vtp
diff --git a/...rizon_estimation/models/mesh/pendulum.vtp → ...rizon_estimation/models/mesh/pendulum.vtp b/...rizon_estimation/models/mesh/pendulum.vtp → ...rizon_estimation/models/mesh/pendulum.vtp
diff --git a/...g_horizon_estimation/multi_cyclic_nmpc.py → ...g_horizon_estimation/multi_cyclic_nmpc.py b/...g_horizon_estimation/multi_cyclic_nmpc.py → ...g_horizon_estimation/multi_cyclic_nmpc.py
diff --git a/...les/muscle_driven_ocp/models/arm26.bioMod → ...les/muscle_driven_ocp/models/arm26.bioMod b/...les/muscle_driven_ocp/models/arm26.bioMod → ...les/muscle_driven_ocp/models/arm26.bioMod
diff --git a/...iven_ocp/models/arm26_with_contact.bioMod → ...iven_ocp/models/arm26_with_contact.bioMod b/...iven_ocp/models/arm26_with_contact.bioMod → ...iven_ocp/models/arm26_with_contact.bioMod
diff --git a/..._driven_ocp/models/mesh/arm_r_humerus.vtp → ..._driven_ocp/models/mesh/arm_r_humerus.vtp b/..._driven_ocp/models/mesh/arm_r_humerus.vtp → ..._driven_ocp/models/mesh/arm_r_humerus.vtp
diff --git a/...cle_driven_ocp/models/mesh/arm_r_ulna.vtp → ...cle_driven_ocp/models/mesh/arm_r_ulna.vtp b/...cle_driven_ocp/models/mesh/arm_r_ulna.vtp → ...cle_driven_ocp/models/mesh/arm_r_ulna.vtp
diff --git a/...le_driven_ocp/models/mesh/ground_ribs.vtp → ...le_driven_ocp/models/mesh/ground_ribs.vtp b/...le_driven_ocp/models/mesh/ground_ribs.vtp → ...le_driven_ocp/models/mesh/ground_ribs.vtp
diff --git a/..._driven_ocp/muscle_activations_tracker.py → ..._driven_ocp/muscle_activations_tracker.py b/..._driven_ocp/muscle_activations_tracker.py → ..._driven_ocp/muscle_activations_tracker.py
diff --git a/..._driven_ocp/muscle_excitations_tracker.py → ..._driven_ocp/muscle_excitations_tracker.py b/..._driven_ocp/muscle_excitations_tracker.py → ..._driven_ocp/muscle_excitations_tracker.py
diff --git a/examples/muscle_driven_ocp/static_arm.py → .../examples/muscle_driven_ocp/static_arm.py b/examples/muscle_driven_ocp/static_arm.py → .../examples/muscle_driven_ocp/static_arm.py
diff --git a/...cle_driven_ocp/static_arm_with_contact.py → ...cle_driven_ocp/static_arm_with_contact.py b/...cle_driven_ocp/static_arm_with_contact.py → ...cle_driven_ocp/static_arm_with_contact.py
diff --git a/...ct_forces_inequality_constraint_muscle.py → ...ct_forces_inequality_constraint_muscle.py b/...ct_forces_inequality_constraint_muscle.py → ...ct_forces_inequality_constraint_muscle.py
diff --git a/...equality_constraint_muscle_excitations.py → ...equality_constraint_muscle_excitations.py b/...equality_constraint_muscle_excitations.py → ...equality_constraint_muscle_excitations.py
diff --git a/...s/2segments_4dof_2contacts_1muscle.bioMod → ...s/2segments_4dof_2contacts_1muscle.bioMod b/...s/2segments_4dof_2contacts_1muscle.bioMod → ...s/2segments_4dof_2contacts_1muscle.bioMod
diff --git a/...ct/muscle_activations_contacts_tracker.py → ...ct/muscle_activations_contacts_tracker.py b/...ct/muscle_activations_contacts_tracker.py → ...ct/muscle_activations_contacts_tracker.py
diff --git a/examples/optimal_time_ocp/models/cube.bioMod → ...mples/optimal_time_ocp/models/cube.bioMod b/examples/optimal_time_ocp/models/cube.bioMod → ...mples/optimal_time_ocp/models/cube.bioMod
diff --git a/...optimal_time_ocp/models/mesh/pendulum.vtp → ...optimal_time_ocp/models/mesh/pendulum.vtp b/...optimal_time_ocp/models/mesh/pendulum.vtp → ...optimal_time_ocp/models/mesh/pendulum.vtp
diff --git a/...s/optimal_time_ocp/models/pendulum.bioMod → ...s/optimal_time_ocp/models/pendulum.bioMod b/...s/optimal_time_ocp/models/pendulum.bioMod → ...s/optimal_time_ocp/models/pendulum.bioMod
diff --git a/...al_time_ocp/multiphase_time_constraint.py → ...al_time_ocp/multiphase_time_constraint.py b/...al_time_ocp/multiphase_time_constraint.py → ...al_time_ocp/multiphase_time_constraint.py
diff --git a/...al_time_ocp/pendulum_min_time_Lagrange.py → ...al_time_ocp/pendulum_min_time_Lagrange.py b/...al_time_ocp/pendulum_min_time_Lagrange.py → ...al_time_ocp/pendulum_min_time_Lagrange.py
diff --git a/...timal_time_ocp/pendulum_min_time_Mayer.py → ...timal_time_ocp/pendulum_min_time_Mayer.py b/...timal_time_ocp/pendulum_min_time_Mayer.py → ...timal_time_ocp/pendulum_min_time_Mayer.py
diff --git a/examples/optimal_time_ocp/time_constraint.py → ...mples/optimal_time_ocp/time_constraint.py b/examples/optimal_time_ocp/time_constraint.py → ...mples/optimal_time_ocp/time_constraint.py
diff --git a/...l_torque_driven_ocp/models/cubeSym.bioMod → ...l_torque_driven_ocp/models/cubeSym.bioMod b/...l_torque_driven_ocp/models/cubeSym.bioMod → ...l_torque_driven_ocp/models/cubeSym.bioMod
diff --git a/...rque_driven_ocp/symmetry_by_constraint.py → ...rque_driven_ocp/symmetry_by_constraint.py b/...rque_driven_ocp/symmetry_by_constraint.py → ...rque_driven_ocp/symmetry_by_constraint.py
diff --git a/..._torque_driven_ocp/symmetry_by_mapping.py → ..._torque_driven_ocp/symmetry_by_mapping.py b/..._torque_driven_ocp/symmetry_by_mapping.py → ..._torque_driven_ocp/symmetry_by_mapping.py
diff --git a/...iven_ocp/maximize_predicted_height_CoM.py → ...iven_ocp/maximize_predicted_height_CoM.py b/...iven_ocp/maximize_predicted_height_CoM.py → ...iven_ocp/maximize_predicted_height_CoM.py
diff --git a/...cp/models/2segments_4dof_2contacts.bioMod → ...cp/models/2segments_4dof_2contacts.bioMod b/...cp/models/2segments_4dof_2contacts.bioMod → ...cp/models/2segments_4dof_2contacts.bioMod
diff --git a/...ocp/models/TruncAnd2Arm_Quaternion.bioMod → ...ocp/models/TruncAnd2Arm_Quaternion.bioMod b/...ocp/models/TruncAnd2Arm_Quaternion.bioMod → ...ocp/models/TruncAnd2Arm_Quaternion.bioMod
diff --git a/...ples/torque_driven_ocp/models/cube.bioMod → ...ples/torque_driven_ocp/models/cube.bioMod b/...ples/torque_driven_ocp/models/cube.bioMod → ...ples/torque_driven_ocp/models/cube.bioMod
diff --git a/..._driven_ocp/models/double_pendulum.bioMod → ..._driven_ocp/models/double_pendulum.bioMod b/..._driven_ocp/models/double_pendulum.bioMod → ..._driven_ocp/models/double_pendulum.bioMod
diff --git a/.../double_pendulum_with_translations.bioMod → .../double_pendulum_with_translations.bioMod b/.../double_pendulum_with_translations.bioMod → .../double_pendulum_with_translations.bioMod
diff --git a/...orque_driven_ocp/models/mass_point.bioMod → ...orque_driven_ocp/models/mass_point.bioMod b/...orque_driven_ocp/models/mass_point.bioMod → ...orque_driven_ocp/models/mass_point.bioMod
diff --git a/...orque_driven_ocp/models/mesh/pendulum.vtp → ...orque_driven_ocp/models/mesh/pendulum.vtp b/...orque_driven_ocp/models/mesh/pendulum.vtp → ...orque_driven_ocp/models/mesh/pendulum.vtp
diff --git a/.../torque_driven_ocp/models/pendulum.bioMod → .../torque_driven_ocp/models/pendulum.bioMod b/.../torque_driven_ocp/models/pendulum.bioMod → .../torque_driven_ocp/models/pendulum.bioMod
diff --git a/...hase_transition_uneven_variable_number.py → ...hase_transition_uneven_variable_number.py b/...hase_transition_uneven_variable_number.py → ...hase_transition_uneven_variable_number.py
diff --git a/examples/torque_driven_ocp/spring_load.py → ...examples/torque_driven_ocp/spring_load.py b/examples/torque_driven_ocp/spring_load.py → ...examples/torque_driven_ocp/spring_load.py
diff --git a/...e_driven_ocp/track_markers_2D_pendulum.py → ...e_driven_ocp/track_markers_2D_pendulum.py b/...e_driven_ocp/track_markers_2D_pendulum.py → ...e_driven_ocp/track_markers_2D_pendulum.py
diff --git a/...cp/track_markers_with_torque_actuators.py → ...cp/track_markers_with_torque_actuators.py b/...cp/track_markers_with_torque_actuators.py → ...cp/track_markers_with_torque_actuators.py
diff --git a/...s/torque_driven_ocp/trampo_quaternions.py → ...s/torque_driven_ocp/trampo_quaternions.py b/...s/torque_driven_ocp/trampo_quaternions.py → ...s/torque_driven_ocp/trampo_quaternions.py
diff --git a/examples/track/models/cube_and_line.bioMod → ...xamples/track/models/cube_and_line.bioMod b/examples/track/models/cube_and_line.bioMod → ...xamples/track/models/cube_and_line.bioMod
diff --git a/examples/track/track_marker_on_segment.py → ...examples/track/track_marker_on_segment.py b/examples/track/track_marker_on_segment.py → ...examples/track/track_marker_on_segment.py
diff --git a/examples/track/track_segment_on_rt.py → ...tim/examples/track/track_segment_on_rt.py b/examples/track/track_segment_on_rt.py → ...tim/examples/track/track_segment_on_rt.py
diff --git a/examples/READM.md b/examples/READM.md
@@ -0,0 +1,2 @@
+Examples can be found in `{ROOT_FOLDER}/bioptim/examples/` or alternatively can be ran directly from the root (or after installing using `python setup.py install`) by running: `python -m bioptim.examples`.
+
diff --git a/tests/test__global_plots.py b/tests/test__global_plots.py
@@ -12,20 +12,19 @@
 from bioptim import OptimalControlProgram, CostType, OdeSolver, Solver
 from bioptim.limits.penalty import PenaltyOption
 
-from .utils import TestUtils
-
 import matplotlib
 
 matplotlib.use("Agg")
 
 
 def test_plot_graphs_one_phase():
     # Load graphs_one_phase
-    bioptim_folder = TestUtils.bioptim_folder()
-    graph = TestUtils.load_module(bioptim_folder + "/examples/torque_driven_ocp/track_markers_with_torque_actuators.py")
+    from bioptim.examples.torque_driven_ocp import track_markers_with_torque_actuators as ocp_module
+
+    bioptim_folder = os.path.dirname(ocp_module.__file__)
 
-    ocp = graph.prepare_ocp(
-        biorbd_model_path=bioptim_folder + "/examples/torque_driven_ocp/models/cube.bioMod",
+    ocp = ocp_module.prepare_ocp(
+        biorbd_model_path=bioptim_folder + "/models/cube.bioMod",
         n_shooting=30,
         final_time=2,
     )
@@ -36,21 +35,22 @@ def test_plot_graphs_one_phase():
 
 def test_plot_merged_graphs():
     # Load graphs_one_phase
-    bioptim_folder = TestUtils.bioptim_folder()
-    merged_graphs = TestUtils.load_module(bioptim_folder + "/examples/muscle_driven_ocp/muscle_excitations_tracker.py")
+    from bioptim.examples.muscle_driven_ocp import muscle_excitations_tracker as ocp_module
+
+    bioptim_folder = os.path.dirname(ocp_module.__file__)
 
     # Define the problem
-    model_path = bioptim_folder + "/examples/muscle_driven_ocp/models/arm26.bioMod"
+    model_path = bioptim_folder + "/models/arm26.bioMod"
     biorbd_model = biorbd.Model(model_path)
     final_time = 0.1
     n_shooting = 5
 
     # Generate random data to fit
     np.random.seed(42)
-    t, markers_ref, x_ref, muscle_excitations_ref = merged_graphs.generate_data(biorbd_model, final_time, n_shooting)
+    t, markers_ref, x_ref, muscle_excitations_ref = ocp_module.generate_data(biorbd_model, final_time, n_shooting)
 
     biorbd_model = biorbd.Model(model_path)  # To prevent from free variable, the model must be reloaded
-    ocp = merged_graphs.prepare_ocp(
+    ocp = ocp_module.prepare_ocp(
         biorbd_model,
         final_time,
         n_shooting,
@@ -67,21 +67,23 @@ def test_plot_merged_graphs():
 
 def test_plot_graphs_multi_phases():
     # Load graphs_one_phase
-    bioptim_folder = TestUtils.bioptim_folder()
-    graphs = TestUtils.load_module(bioptim_folder + "/examples/getting_started/example_multiphase.py")
-    ocp = graphs.prepare_ocp(biorbd_model_path=bioptim_folder + "/examples/getting_started/models/cube.bioMod")
+    from bioptim.examples.getting_started import example_multiphase as ocp_module
+
+    bioptim_folder = os.path.dirname(ocp_module.__file__)
+
+    ocp = ocp_module.prepare_ocp(biorbd_model_path=bioptim_folder + "/models/cube.bioMod")
     sol = ocp.solve()
     sol.graphs(automatically_organize=False)
 
 
 def test_add_new_plot():
     # Load graphs_one_phase
-    bioptim_folder = TestUtils.bioptim_folder()
-    graphs = TestUtils.load_module(
-        bioptim_folder + "/examples/torque_driven_ocp/track_markers_with_torque_actuators.py"
-    )
-    ocp = graphs.prepare_ocp(
-        biorbd_model_path=bioptim_folder + "/examples/torque_driven_ocp/models/cube.bioMod",
+    from bioptim.examples.torque_driven_ocp import track_markers_with_torque_actuators as ocp_module
+
+    bioptim_folder = os.path.dirname(ocp_module.__file__)
+
+    ocp = ocp_module.prepare_ocp(
+        biorbd_model_path=bioptim_folder + "/models/cube.bioMod",
         n_shooting=20,
         final_time=0.5,
     )
@@ -124,9 +126,11 @@ def test_add_new_plot():
 
 def test_console_objective_functions():
     # Load graphs_one_phase
-    bioptim_folder = TestUtils.bioptim_folder()
-    graphs = TestUtils.load_module(bioptim_folder + "/examples/getting_started/example_multiphase.py")
-    ocp = graphs.prepare_ocp(biorbd_model_path=bioptim_folder + "/examples/getting_started/models/cube.bioMod")
+    from bioptim.examples.getting_started import example_multiphase as ocp_module
+
+    bioptim_folder = os.path.dirname(ocp_module.__file__)
+
+    ocp = ocp_module.prepare_ocp(biorbd_model_path=bioptim_folder + "/models/cube.bioMod")
     sol = ocp.solve()
     ocp = sol.ocp  # We will override ocp with known and controlled values for the test
 
@@ -206,3 +210,4 @@ def override_penalty(pen: list[PenaltyOption]):
 
     sys.stdout = sys.__stdout__  # Reset redirect.
     assert captured_output.getvalue() == expected_output
+
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		Examples can be found in `{ROOT_FOLDER}/bioptim/examples/` or alternatively can be ran directly from the root (or after installing using `python setup.py install`) by running: `python -m bioptim.examples`.