Extend the smoke test to include screen capture

Author: jonasschneider

atari_py/tests/test_smoke.py

@@ -1,8 +1,16 @@
 import atari_py
+import numpy as np
 
 def test_smoke():
     pong_path = atari_py.get_game_path('pong')
     ale = atari_py.ALEInterface()
     ale.loadROM(pong_path)
     action_set = ale.getMinimalActionSet()
+
+    # Test stepping
     ale.act(action_set[0])
+
+    # Test screen capture
+    (screen_width,screen_height) = ale.getScreenDims()
+    arr = np.zeros((screen_height, screen_width, 4), dtype=np.uint8)
+    ale.getScreenRGB(arr)

tox.ini

@@ -8,6 +8,7 @@ envlist = py27, py35
 
 [testenv:py35]
 whitelist_externals=make
+                    echo
 install_command=echo {packages}
 deps =
     nose2