Browse Source

Added check in run_tests.py

Maxime Chevalier-Boisvert 7 years ago
parent
commit
95448c1ebd
1 changed files with 6 additions and 0 deletions
  1. 6 0
      run_tests.py

+ 6 - 0
run_tests.py

@@ -35,6 +35,12 @@ for envName in sorted(envSet):
         assert reward >= env.reward_range[0], reward
         assert reward >= env.reward_range[0], reward
         assert reward <= env.reward_range[1], reward
         assert reward <= env.reward_range[1], reward
 
 
+        if done:
+            env.reset()
+
+            # Check that the agent doesn't overlap with an object
+            assert env.grid.get(*env.agentPos) is None
+
         env.render('rgb_array')
         env.render('rgb_array')
 
 
     env.close()
     env.close()