def test_update_completed_trial(random_dt):
"""Successfully push a completed trial into database."""
with OrionState(trials=generate_trials(['new'])) as cfg:
exp = Experiment('supernaekei')
exp._id = cfg.trials[0]['experiment']
trial = exp.reserve_trial()
results_file = tempfile.NamedTemporaryFile(mode='w',
prefix='results_',
suffix='.log',
dir='.',
delete=True)
# Generate fake result
with open(results_file.name, 'w') as file:
json.dump([{
'name': 'loss',
'type': 'objective',
'value': 2
}], file)
# --
exp.update_completed_trial(trial, results_file=results_file)
yo = get_storage().fetch_trials(exp)[0].to_dict()
assert len(yo['results']) == len(trial.results)
assert yo['results'][0] == trial.results[0].to_dict()
assert yo['status'] == 'completed'
assert yo['end_time'] == random_dt
results_file.close()
def test_update_completed_trial(random_dt):
"""Successfully push a completed trial into database."""
with OrionState(trials=generate_trials(["new"])) as cfg:
exp = Experiment("supernaekei", mode="x")
exp._id = cfg.trials[0]["experiment"]
trial = exp.reserve_trial()
results_file = tempfile.NamedTemporaryFile(mode="w",
prefix="results_",
suffix=".log",
dir=".",
delete=True)
# Generate fake result
with open(results_file.name, "w") as file:
json.dump([{
"name": "loss",
"type": "objective",
"value": 2
}], file)
# --
exp.update_completed_trial(trial, results_file=results_file)
yo = get_storage().fetch_trials(exp)[0].to_dict()
assert len(yo["results"]) == len(trial.results)
assert yo["results"][0] == trial.results[0].to_dict()
assert yo["status"] == "completed"
assert yo["end_time"] == random_dt
results_file.close()
