Diff of /tests/test.simbody.py [000000] .. [077a87]

Switch to side-by-side view

--- a
+++ b/tests/test.simbody.py
@@ -0,0 +1,27 @@
+from osim.env import ProstheticsEnv
+import numpy as np
+import unittest
+
+class SimulationTest(unittest.TestCase):
+    def test1(self):
+        
+        env = ProstheticsEnv(visualize=True)
+        observation = env.reset()
+
+        simbody_state = env.osim_model.get_state()
+        print(simbody_state.getNumSubsystems())
+        print(simbody_state.getY())
+        oldy = simbody_state.updY()
+        for i in range(len(oldy)):
+            oldy[i] += 0.2
+        print(simbody_state.getY())
+        env.osim_model.set_state(simbody_state)
+
+        action = env.action_space.sample()
+
+        for i in range(50):
+            env.step(action)
+
+
+if __name__ == '__main__':
+    unittest.main()