def test_run_multi_fcnet_td3_independent(self): # Run the script; verify it executes without failure. args = parse_train_options( '', '', args=[ "multiagent-ring-v0", "--initial_exploration_steps", "1", "--total_steps", "500", "--log_interval", "500", ], multiagent=True, hierarchical=False, ) run_multi_fcnet(args, 'data/multi-fcnet') # Check that the folders were generated. self.assertTrue( os.path.isdir( os.path.join(os.getcwd(), "data/multi-fcnet/multiagent-ring-v0"))) # Clear anything that was generated. shutil.rmtree(os.path.join(os.getcwd(), "data"))
def test_run_multi_fcnet_td3_maddpg_shared(self): # Run the script; verify it executes without failure. args = parse_options('', '', args=[ "multiagent-ring_small", "--shared", "--maddpg", "--initial_exploration_steps", "1", "--total_steps", "500", "--log_interval", "500", ]) run_multi_fcnet(args, 'data/multi-fcnet') # Check that the folders were generated. self.assertTrue( os.path.isdir( os.path.join(os.getcwd(), "data/multi-fcnet/multiagent-ring_small"))) # Clear anything that was generated. shutil.rmtree(os.path.join(os.getcwd(), "data"))
def test_run_multi_fcnet_td3_independent(self): # Run the script; verify it executes without failure. args = parse_options( '', '', args=["multiagent-ring0", "--total_steps", "2000"]) run_multi_fcnet(args, 'data/multi-fcnet') # Check that the folders were generated. self.assertTrue( os.path.isdir( os.path.join(os.getcwd(), "data/multi-fcnet/multiagent-ring0"))) # Clear anything that was generated. shutil.rmtree(os.path.join(os.getcwd(), "data"))