--- a +++ b/tests/test.clipping.py @@ -0,0 +1,16 @@ +from osim.env import L2RunEnv +import numpy as np +import unittest + +class ActivationsTest(unittest.TestCase): + def test_clipping(self): + env = L2RunEnv(visualize=False) + observation = env.reset() + + env.step(np.array([5.0] * 18)) + self.assertLessEqual( np.sum(env.osim_model.last_action), 18.1 ) + env.step(np.array([-1.0] * 18)) + self.assertGreaterEqual( np.sum(env.osim_model.last_action), -0.1 ) + +if __name__ == '__main__': + unittest.main()