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)