tests.test_optimizers._test_max_time

Complexity

Total Complexity

3

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	objective_function()	0	3	1
A	test_max_time_0()	0	8	1
A	test_max_time_1()	0	8	1

import time
import pytest
import numpy as np
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import cross_val_score
from sklearn.tree import DecisionTreeClassifier

from ._parametrize import optimizers


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


search_space = {
    "x1": np.arange(0, 100, 1),  # small search space because of smbo
}


@pytest.mark.parametrize(*optimizers)
def test_max_time_0(Optimizer):
    c_time1 = time.time()
    opt = Optimizer(search_space)
    opt.search(objective_function, n_iter=1000000, max_time=0.1)
    diff_time1 = time.time() - c_time1

    assert diff_time1 < 1


@pytest.mark.parametrize(*optimizers)
def test_max_time_1(Optimizer):
    c_time1 = time.time()
    opt = Optimizer(search_space)
    opt.search(objective_function, n_iter=1000000, max_time=1)
    diff_time1 = time.time() - c_time1

    assert 0.3 < diff_time1 < 2