for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
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
def test_max_time_1(Optimizer):
opt.search(objective_function, n_iter=1000000, max_time=1)
assert 0.3 < diff_time1 < 2