tests.test_performance.test_global_opt.test_global_perf()

Complexity

Conditions

2

Size

Total Lines	27
Code Lines	20

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import pytest
@pytest.mark.parametrize(*opt_global_l)
def test_global_perf(Optimizer):
    ackley_function = RastriginFunction(n_dim=1, metric="score")

    search_space = {"x0": np.arange(-100, 101, 1)}
    initialize = {"vertices": 2}

    n_opts = 33
    n_iter = 100

    scores = []
    for rnd_st in tqdm(range(n_opts)):
        opt = Optimizer(search_space, initialize=initialize)
        opt.search(
            ackley_function,
            n_iter=n_iter,
            random_state=rnd_st,
            memory=False,
            verbosity=False,
        )

        scores.append(opt.best_score)
    score_mean = np.array(scores).mean()

    print("\n score_mean", score_mean)

    assert score_mean > -5