def test_bayesian_oracle_maximize(tmp_dir):
    hps = hp_module.HyperParameters()
    hps.Int('a', (- 100), 100)
    oracle = bo_module.BayesianOptimizationOracle(objective=kt.Objective('score', direction='max'), max_trials=20, hyperparameters=hps, num_initial_points=2)
    oracle._set_project_dir(tmp_dir, 'untitled')
    for i in range(5):
        trial = trial_module.Trial(hyperparameters=hps.copy())
        trial.hyperparameters.values['a'] = (10 * i)
        trial.score = i
        trial.status = 'COMPLETED'
        oracle.trials[trial.trial_id] = trial
    for i in range(5):
        trial = trial_module.Trial(hyperparameters=hps.copy())
        trial.hyperparameters.values['a'] = ((- 10) * i)
        trial.score = (- i)
        trial.status = 'COMPLETED'
        oracle.trials[trial.trial_id] = trial
    trial = oracle.create_trial('tuner0')
    assert (trial.status == 'RUNNING')
    assert (trial.hyperparameters.get('a') > 0)
def test_bayesian_oracle_maximize(tmp_dir):
    hps = hp_module.HyperParameters()
    hps.Int('a', (- 100), 100)
    oracle = bo_module.BayesianOptimizationOracle(objective=kt.Objective('score', direction='max'), max_trials=20, hyperparameters=hps, num_initial_points=2)
    oracle._set_project_dir(tmp_dir, 'untitled')
    for i in range(5):
        trial = trial_module.Trial(hyperparameters=hps.copy())
        trial.hyperparameters.values['a'] = (10 * i)
        trial.score = i
        trial.status = 'COMPLETED'
        oracle.trials[trial.trial_id] = trial
    for i in range(5):
        trial = trial_module.Trial(hyperparameters=hps.copy())
        trial.hyperparameters.values['a'] = ((- 10) * i)
        trial.score = (- i)
        trial.status = 'COMPLETED'
        oracle.trials[trial.trial_id] = trial
    trial = oracle.create_trial('tuner0')
    assert (trial.status == 'RUNNING')
    assert (trial.hyperparameters.get('a') > 0)