gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

tests: minor fix for functional tests (#289)

Browse Source 
* tests: minor fix for functional tests

* tests: fix value
This commit is contained in:
anabwan
2019-04-15 12:28:23 +03:00
committed by GitHub
parent 88f9c926ab
commit 20a8dea0dd
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/tests/test_checkpoint.py
View File

@@ -91,7 +91,7 @@ def test_restore_checkpoint(preset_args, clres, framework,
return p
if framework == "mxnet":
preset_args = Def.Presets.mxnet_args_test
preset_args = Def.Presets.mxnet_args_test[0]
p_valid_params = p_utils.validation_params(preset_args)
create_cp_proc = _create_cmd_and_run(flag=['--checkpoint_save_secs', '5'])

4
rl_coach/tests/utils/args_utils.py
View File

@@ -395,6 +395,10 @@ def validate_arg_result(flag, p_valid_params, clres=None, process=None,
results.append(last_step[-1])
time.sleep(1)
# get the first value after heat-up
time.sleep(1)
results.append(csv["Total steps"].values[-1])
assert int(results[-1]) >= Def.Consts.num_hs, \
Def.Consts.ASSERT_MSG.format("bigger than " +
str(Def.Consts.num_hs), results[-1])