temp save

hanruihua · Nov 27, 2024 · db36053 · db36053
1 parent 27b7504
commit db36053
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 45 deletions.
diff --git a/irsim/test_collision_avoidance.py b/irsim/test_collision_avoidance.py
diff --git a/irsim/test_keyboard_control.py b/irsim/test_keyboard_control.py
diff --git a/tests/test_all_objects.py b/tests/test_all_objects.py
@@ -1,30 +1,29 @@
 import irsim
 
-def test_collision_avoidance():
+def test_all_objects():
 
     env = irsim.make("test_collision_avoidance.yaml", save_ani=False, full=False, display=False)
 
-    for i in range(100):
-
+    for i in range(10):
         env.step()
         env.render(0.01)
-
+        
     env.end()
-
-
-def test_all_objects():
-
+
     env = irsim.make('test_all_objects.yaml', display=False)
 
-    for i in range(100):
-
+    for i in range(10):
         env.step()
         env.render(0.01)
-        
+    env.end()
 
+    env = irsim.make('test_keyboard_control.yaml', save_ani=False, display=False)
+
+    for i in range(10):
+        env.step()
+        env.render(0.01)
     env.end()
 
 
 if __name__ == "__main__":
-    test_collision_avoidance()
     test_all_objects()
diff --git a/irsim/test_keyboard_control.yaml → tests/test_keyboard_control.yaml b/irsim/test_keyboard_control.yaml → tests/test_keyboard_control.yaml