reorganize tests

Author: kengz

test/test_atari.py

@@ -0,0 +1,23 @@
+import unittest
+import pytest
+from os import environ
+from rl.experiment import run
+from . import conftest
+import pandas as pd
+
+
+class AtariTest(unittest.TestCase):
+
+    @unittest.skipIf(environ.get('CI'),
+                     "Delay CI test until dev stable")
+    @classmethod
+    def test_breakout_dqn(cls):
+        data_df = run('breakout_dqn')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @unittest.skipIf(environ.get('CI'),
+                     "Delay CI test until dev stable")
+    @classmethod
+    def test_breakout_double_dqn(cls):
+        data_df = run('breakout_double_dqn')
+        assert isinstance(data_df, pd.DataFrame)

test/test_box2d.py

@@ -0,0 +1,24 @@
+import unittest
+import pytest
+from os import environ
+from rl.experiment import run
+from . import conftest
+import pandas as pd
+
+
+class Box2DTest(unittest.TestCase):
+
+    @classmethod
+    def test_lunar_dqn(cls):
+        data_df = run('lunar_dqn')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_lunar_double_dqn(cls):
+        data_df = run('lunar_double_dqn')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_walker_ddpg_linearnoise(cls):
+        data_df = run('walker_ddpg_linearnoise')
+        assert isinstance(data_df, pd.DataFrame)

test/test_advanced.py test/test_classic.pytest/test_advanced.py renamed to test/test_classic.py

@@ -6,58 +6,64 @@
 import pandas as pd
 
 
-class AdvancedTest(unittest.TestCase):
+class ClassicTest(unittest.TestCase):
 
     @classmethod
-    def test_sarsa(cls):
-        data_df = run('sarsa')
+    def test_quickstart_dqn(cls):
+        data_df = run('quickstart_dqn')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_exp_sarsa(cls):
-        data_df = run('exp_sarsa')
+    def test_dqn_epsilon(cls):
+        data_df = run('dqn_epsilon')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_offpol_sarsa(cls):
-        data_df = run('offpol_sarsa')
+    def test_dqn(cls):
+        data_df = run('dqn')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_acrobot(cls):
-        data_df = run('acrobot')
+    def test_double_dqn(cls):
+        data_df = run('double_dqn')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_mountain_dqn(cls):
-        data_df = run('mountain_dqn')
+    def test_sarsa(cls):
+        data_df = run('sarsa')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_exp_sarsa(cls):
+        data_df = run('exp_sarsa')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_lunar_dqn(cls):
-        data_df = run('lunar_dqn')
+    def test_offpol_sarsa(cls):
+        data_df = run('offpol_sarsa')
         assert isinstance(data_df, pd.DataFrame)
 
-    @unittest.skipIf(environ.get('CI'),
-                     "Delay CI test until dev stable")
     @classmethod
-    def test_breakout_dqn(cls):
-        data_df = run('breakout_dqn')
+    def test_cartpole_ac_argmax(cls):
+        data_df = run('cartpole_ac_argmax')
         assert isinstance(data_df, pd.DataFrame)
 
-    @unittest.skipIf(environ.get('CI'),
-                     "Delay CI test until dev stable")
     @classmethod
-    def test_breakout_double_dqn(cls):
-        data_df = run('breakout_double_dqn')
+    def test_dqn_v1(cls):
+        data_df = run('dqn_v1')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_cartpole_ac_argmax(cls):
-        data_df = run('cartpole_ac_argmax')
+    def test_acrobot(cls):
+        data_df = run('acrobot')
         assert isinstance(data_df, pd.DataFrame)
 
     @classmethod
-    def test_pendulum_ddpg(cls):
-        data_df = run('pendulum_ddpg')
+    def test_pendulum_ddpg_linearnoise(cls):
+        data_df = run('pendulum_ddpg_linearnoise')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_mountain_dqn(cls):
+        data_df = run('mountain_dqn')
         assert isinstance(data_df, pd.DataFrame)

test/test_basic.py test/test_dev.pytest/test_basic.py renamed to test/test_dev.py

@@ -6,7 +6,7 @@
 import pandas as pd
 
 
-class BasicTest(unittest.TestCase):
+class DevTest(unittest.TestCase):
 
     @classmethod
     def test_clean_import(cls):
@@ -46,13 +46,3 @@ def test_dqn_pass(cls):
     # def test_dqn_random_search(cls):
     #     data_df = run('test_dqn_random_search', param_selection=True)
     #     assert isinstance(data_df, pd.DataFrame)
-
-    @classmethod
-    def test_dqn(cls):
-        data_df = run('dqn')
-        assert isinstance(data_df, pd.DataFrame)
-
-    @classmethod
-    def test_double_dqn(cls):
-        data_df = run('double_dqn')
-        assert isinstance(data_df, pd.DataFrame)