gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00
Code Issues Projects Releases Wiki Activity

tests: stabling functional tests (#355)

Browse Source 
* tests: stabling functional tests

* functional removed
This commit is contained in:
anabwan
2019-06-20 15:30:47 +03:00
committed by GitHub
parent 8e812ef82f
commit 7b5d6a3f03
2 changed files with 10 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rl_coach/tests/test_checkpoint.py
View File

@@ -125,11 +125,6 @@ def test_restore_checkpoint(preset_args, clres, framework,
# send CTRL+C to close experiment
create_cp_proc.send_signal(signal.SIGINT)
csv = pd.read_csv(csv_list[0])
rewards = csv['Evaluation Reward'].values
rewards = rewards[~np.isnan(rewards)]
max_reward = np.amax(rewards)
if os.path.isdir(checkpoint_dir):
shutil.copytree(exp_dir, checkpoint_test_dir)
shutil.rmtree(exp_dir)
@@ -146,8 +141,9 @@ def test_restore_checkpoint(preset_args, clres, framework,
csv = pd.read_csv(new_csv_list[0])
res = csv['Episode Length'].values[-1]
assert res == max_reward, Def.Consts.ASSERT_MSG.format(str(max_reward),
str(res))
expected_reward = 100
assert res >= expected_reward, Def.Consts.ASSERT_MSG.format(
str(expected_reward), str(res))
restore_cp_proc.kill()
test_folder = os.path.join(Def.Path.experiments, Def.Path.test_dir)