tests.test_optimizers.test_parameter._base_para_test

Complexity

Total Complexity

2

Size/Duplication

Total Lines	30
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	objective_function()	0	3	1
A	_base_para_test_func()	0	14	1

# Author: Simon Blanke
# Email: simon.blanke@yahoo.com
# License: MIT License

import numpy as np


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


search_space = {"x1": np.arange(-100, 101, 1)}


def _base_para_test_func(opt_para, optimizer):
    opt = optimizer(search_space, **opt_para)
    opt.search(
        objective_function,
        n_iter=30,
        memory=False,
        verbosity=False,
        initialize={"vertices": 1},
    )

    para_key = list(opt_para.keys())[0]
    para_value = getattr(opt, para_key)

    assert para_value == opt_para[para_key]