浏览代码

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