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()