瀏覽代碼

Added test

Maxime Chevalier-Boisvert 7 年之前
父節點
當前提交
bb6dc6196e
共有 1 個文件被更改,包括 9 次插入1 次删除
  1. 9 1
      run_tests.py

+ 9 - 1
run_tests.py

@@ -22,7 +22,15 @@ for envName in sorted(envSet):
     env.reset()
     env.render('rgb_array')
 
-    env.seed()
+    # Verify that the same seed always produces the same environment
+    for i in range(0, 5):
+        seed = 1337 + i
+        env.seed(seed)
+        grid1 = env.grid.encode()
+        env.seed(seed)
+        grid2 = env.grid.encode()
+        assert np.array_equal(grid2, grid1)
+
     env.reset()
 
     # Run for a few episodes