run_tests.py 467 B

12345678910111213141516171819202122
  1. #!/usr/bin/env python3
  2. import random
  3. import gym
  4. from gym_minigrid.register import envSet
  5. print('%d environments registered' % len(envSet))
  6. for envName in sorted(envSet):
  7. #print(envName)
  8. # Load the gym environment
  9. env = gym.make(envName)
  10. env.reset()
  11. env.render('rgb_array')
  12. numActions = env.action_space.n
  13. for i in range(500):
  14. action = random.randint(0, numActions - 1)
  15. env.step(action)
  16. env.render('rgb_array')