From 0b50a69c3c25ff30a212f03876bb028982a2672f Mon Sep 17 00:00:00 2001 From: kengz Date: Wed, 19 Apr 2017 00:03:57 -0400 Subject: [PATCH] add ac, ddpg tests --- test/test_advanced.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/test/test_advanced.py b/test/test_advanced.py index eec9607..7407e56 100644 --- 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)