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

Switch to side-by-side view

--- a
+++ b/tests/test.env.py
@@ -0,0 +1,24 @@
+from osim.env import L2RunEnv
+import numpy as np
+import unittest
+
+class SimulationTest(unittest.TestCase):
+    def test_reset(self):
+        env = L2RunEnv(visualize=False)
+        for i in range(10):
+            observation = env.reset()
+
+        action = env.action_space.sample()
+        action[5] = np.NaN
+        self.assertRaises(ValueError, env.step, action)
+
+    def test_actions(self):
+        env = L2RunEnv(visualize=False)
+        env.reset()
+        v = env.action_space.sample()
+        v[0] = 1.5
+        v[1] = -0.5
+        observation, reward, done, info = env.step(v)
+
+if __name__ == '__main__':
+    unittest.main()