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