소스 검색

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