add ac, ddpg tests

kengz · Apr 19, 2017 · 0b50a69 · 0b50a69
1 parent be6c2a9
commit 0b50a69
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/test_advanced.py b/test/test_advanced.py
@@ -51,3 +51,13 @@ def test_breakout_dqn(cls):
     def test_breakout_double_dqn(cls):
         data_df = run('breakout_double_dqn')
         assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_cartpole_ac_argmax(cls):
+        data_df = run('cartpole_ac_argmax')
+        assert isinstance(data_df, pd.DataFrame)
+
+    @classmethod
+    def test_pendulum_ddpg(cls):
+        data_df = run('pendulum_ddpg')
+        assert isinstance(data_df, pd.DataFrame)