tests.local_test_performance.local_test_global_opt

Complexity

Total Complexity

2

Size/Duplication

Total Lines	54
Duplicated Lines	55.56 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	test_global_perf()	30	30	2

import pytest
from tqdm import tqdm
import numpy as np

from surfaces.test_functions.mathematical import RastriginFunction

from gradient_free_optimizers import (
    RandomSearchOptimizer,
    RandomRestartHillClimbingOptimizer,
    RandomAnnealingOptimizer,
)


opt_global_l = (
    "Optimizer",
    [
        (RandomSearchOptimizer),
        (RandomRestartHillClimbingOptimizer),
        (RandomAnnealingOptimizer),
    ],
)


@pytest.mark.parametrize(*opt_global_l)

This code seems to be duplicated in your project.

def test_global_perf(Optimizer):
    ackley_function = RastriginFunction(n_dim=1, metric="score")

    def objective_function(para):
        score = -para["x1"] * para["x1"]
        return score

    search_space = {"x1": 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, random_state=rnd_st)
        opt.search(
            objective_function,
            n_iter=n_iter,
            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