tests.test_checks

Complexity

Total Complexity

38

Size/Duplication

Total Lines	232
Duplicated Lines	0 %

Importance

Changes

0

20 Functions

Rating	Name	Duplication	Size	Complexity
A	test_n_jobs_value()	0	9	2
A	test_n_iter_key()	0	9	2
A	test_distribution_value()	0	9	2
A	test_search_space_key()	0	8	2
A	_base_test()	0	4	1
A	test_n_iter_value()	0	9	2
A	test_search_space_value()	0	8	2
A	test_init_para_key()	0	9	2
A	test_function_parameter_value()	0	8	2
A	test_n_jobs_key()	0	9	2
A	test_init_para_value()	0	9	2
A	test_distribution_key()	0	9	2
A	test_optimizer_key()	0	9	2
A	test_optimizer_value()	0	9	2
A	test_objective_function_value()	0	8	2
A	objective_function()	0	9	1
A	test_memory_value()	0	9	2
A	test_function_parameter_key()	0	8	2
A	test_memory_key()	0	9	2
A	test_objective_function_key()	0	8	2

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

import pytest
import numpy as np

from sklearn.datasets import load_iris
from sklearn.model_selection import cross_val_score
from sklearn.tree import DecisionTreeClassifier

from hyperactive import Optimizer

data = load_iris()
X, y = data.data, data.target


def objective_function(para):
    dtc = DecisionTreeClassifier(
        max_depth=para["max_depth"],
        min_samples_split=para["min_samples_split"],
        min_samples_leaf=para["min_samples_leaf"],
    )
    scores = cross_val_score(dtc, para["features"], para["target"], cv=2)

    return scores.mean()


search_space = {
    "max_depth": range(1, 21),
    "min_samples_split": range(2, 21),
    "min_samples_leaf": range(1, 21),
}


def _base_test(search, opt_args={}, time=None):
    opt = Optimizer(**opt_args)
    opt.add_search(**search)
    opt.run(time)


def test_objective_function_key():
    search = {
        "objective_function_": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_objective_function_value():
    search = {
        "objective_function": 1,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_function_parameter_key():
    search = {
        "objective_function": objective_function,
        "function_parameter_": {"features": X, "target": y},
        "search_space": search_space,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_function_parameter_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": 1,
        "search_space": search_space,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_search_space_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space_": search_space,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_search_space_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": 1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_memory_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "memory_": "short",
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_memory_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "memory": 1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_optimizer_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "optimizer_": 1,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_optimizer_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "optimizer": 1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_n_iter_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "n_iter_": 10,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_n_iter_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "n_iter": 0.1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_n_jobs_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "n_jobs_": 1,
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_n_jobs_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "n_jobs": 0.1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_init_para_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "init_para_": {},
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_init_para_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "init_para": 1,
    }
    with pytest.raises(ValueError):
        _base_test(search)


def test_distribution_key():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "distribution_": {},
    }
    with pytest.raises(TypeError):
        _base_test(search)


def test_distribution_value():
    search = {
        "objective_function": objective_function,
        "function_parameter": {"features": X, "target": y},
        "search_space": search_space,
        "distribution": 1,
    }
    with pytest.raises(ValueError):
        _base_test(search)