Add test for restoring from state multiple times

IntrovertInDisguise · Oct 18, 2021 · 2d72af0 · 2d72af0
1 parent ab6d9b6
commit 2d72af0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/pybullet/test_util.py b/tests/pybullet/test_util.py
@@ -46,6 +46,8 @@ def _get_and_set_state(env_name):
     env.reset()
     state = handler.get_current_state(env)
     handler.set_env_state(state, env)
+    # test if we can restore the state multiple times
+    handler.set_env_state(state, env)
 
 
 def test_freeze():