file for speed testing

Kautenja · Jan 21, 2019 · 35b7dbb · 35b7dbb
1 parent d0d648c
commit 35b7dbb
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/speedtest.py b/speedtest.py
@@ -0,0 +1,15 @@
+from nes_py import NESEnv
+import tqdm
+env = NESEnv('./nes_py/tests/games/super-mario-bros-1.nes')
+
+done = True
+
+try:
+    for _ in tqdm.tqdm(range(5000)):
+        if done:
+            state = env.reset()
+            done = False
+        else:
+            state, reward, done, info = env.step(env.action_space.sample())
+except KeyboardInterrupt:
+    pass