def test_register_trials(random_dt):
"""Register a list of newly proposed trials/parameters."""
with OrionState():
exp = Experiment('supernaekei')
exp._id = 0
trials = [
Trial(params=[{
'name': 'a',
'type': 'integer',
'value': 5
}]),
Trial(params=[{
'name': 'b',
'type': 'integer',
'value': 6
}]),
]
for trial in trials:
exp.register_trial(trial)
yo = list(
map(lambda trial: trial.to_dict(),
get_storage().fetch_trials(exp)))
assert len(yo) == len(trials)
assert yo[0]['params'] == list(
map(lambda x: x.to_dict(), trials[0]._params))
assert yo[1]['params'] == list(
map(lambda x: x.to_dict(), trials[1]._params))
assert yo[0]['status'] == 'new'
assert yo[1]['status'] == 'new'
assert yo[0]['submit_time'] == random_dt
assert yo[1]['submit_time'] == random_dt
def test_register_trials(random_dt):
"""Register a list of newly proposed trials/parameters."""
with OrionState():
exp = Experiment("supernaekei", mode="x")
exp._id = 0
trials = [
Trial(params=[{
"name": "a",
"type": "integer",
"value": 5
}]),
Trial(params=[{
"name": "b",
"type": "integer",
"value": 6
}]),
]
for trial in trials:
exp.register_trial(trial)
yo = list(
map(lambda trial: trial.to_dict(),
get_storage().fetch_trials(exp)))
assert len(yo) == len(trials)
assert yo[0]["params"] == list(
map(lambda x: x.to_dict(), trials[0]._params))
assert yo[1]["params"] == list(
map(lambda x: x.to_dict(), trials[1]._params))
assert yo[0]["status"] == "new"
assert yo[1]["status"] == "new"
assert yo[0]["submit_time"] == random_dt
assert yo[1]["submit_time"] == random_dt
