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()]