Add benchmark for full simulation step

ami-iit · Feb 27, 2025 · cafe87c · cafe87c
1 parent ceeb26e
commit cafe87c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -132,3 +132,15 @@ def test_relaxed_rigid_contact_model(
         model.contact_model = jaxsim.rbda.contacts.RelaxedRigidContacts()
 
     benchmark_test_function(js.ode.system_dynamics, model, benchmark, batch_size)
+
+
+@pytest.mark.benchmark
+def test_simulation_step(
+    jaxsim_model_ergocub_reduced: js.model.JaxSimModel, benchmark, batch_size
+):
+    model = jaxsim_model_ergocub_reduced
+
+    with model.editable(validate=False) as model:
+        model.contact_model = jaxsim.rbda.contacts.RelaxedRigidContacts()
+
+    benchmark_test_function(js.model.step, model, benchmark, batch_size)