Преглед на файлове

Added check in run_tests.py

Maxime Chevalier-Boisvert преди 7 години
родител
ревизия
95448c1ebd
променени са 1 файла, в които са добавени 6 реда и са изтрити 0 реда
  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[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.close()