Browse Source

Added test

Maxime Chevalier-Boisvert 7 years ago
parent
commit
bb6dc6196e
1 changed files with 9 additions and 1 deletions
  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