def test_fetch_all_trials():
"""Fetch a list of all trials"""
with OrionState(trials=generate_trials(["new", "reserved", "completed"])) as cfg:
exp = Experiment("supernaekei", mode="x")
exp._id = cfg.trials[0]["experiment"]
trials = list(map(lambda trial: trial.to_dict(), exp.fetch_trials({})))
assert trials == cfg.trials
def test_fetch_all_trials():
"""Fetch a list of all trials"""
with OrionState(
trials=generate_trials(['new', 'reserved', 'completed'])) as cfg:
exp = Experiment('supernaekei')
exp._id = cfg.trials[0]['experiment']
trials = list(map(lambda trial: trial.to_dict(), exp.fetch_trials({})))
assert trials == cfg.trials
def test_experiment_pickleable():
"""Test experiment instance is pickleable"""
with OrionState(trials=generate_trials(["new"])) as cfg:
exp = Experiment("supernaekei", mode="x")
exp._id = cfg.trials[0]["experiment"]
exp_trials = exp.fetch_trials()
assert len(exp_trials) > 0
from orion.storage.base import storage_factory
exp_bytes = pickle.dumps(exp)
new_exp = pickle.loads(exp_bytes)
assert [trial.to_dict() for trial in exp_trials
] == [trial.to_dict() for trial in new_exp.fetch_trials()]
