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