test: add more parameterized number of rigid bodies into test

sy-cui · Sep 7, 2023 · 854898d · 854898d
1 parent f9d83d3
commit 854898d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modules/test_memory_block_rigid_body.py b/tests/test_modules/test_memory_block_rigid_body.py
@@ -65,7 +65,7 @@ def test_construct_memory_block_structures_for_rigid_bodies(n_bodies):
     assert issubclass(memory_block_list[0].__class__, MemoryBlockRigidBody)
 
 
-@pytest.mark.parametrize("n_bodies", [5])
+@pytest.mark.parametrize("n_bodies", [1, 2, 5, 6])
 def test_memory_block_rigid_body(n_bodies):
     """
     Test memory block logic for rigid bodies.