Example #1
0
def test_global_minimum():
    for elem_X in INFO_FUN.get('global_minimum_X', np.array([])):
        cur_X = utils_benchmarks.get_covariate(INFO_FUN, elem_X, INT_DIM)
        val_fun = FUN_TARGET(np.expand_dims(elem_X, axis=0))
        assert (np.abs(val_fun - INFO_FUN.get('global_minimum_y', np.inf)) <
                TEST_EPSILON).all()
def test_get_covariate():
    info_branin = copy.deepcopy(benchmarks.INFO_BRANIN)
    info_ackley = copy.deepcopy(benchmarks.INFO_ACKLEY)
    with pytest.raises(AssertionError) as error:
        utils_benchmarks.get_covariate(info_branin, np.array([0.0, 0.0]), 2.1)
    with pytest.raises(AssertionError) as error:
        utils_benchmarks.get_covariate(info_branin, 'abc', 2)
    with pytest.raises(AssertionError) as error:
        utils_benchmarks.get_covariate(1, np.array([0.0, 0.0]), 2)
    with pytest.raises(AssertionError) as error:
        utils_benchmarks.get_covariate(info_branin, np.array([[0.0, 0.0]]), 2)
    with pytest.raises(ValueError) as error:
        utils_benchmarks.get_covariate(info_branin, np.array([0.0, 0.0, 0.0]),
                                       2)
    with pytest.raises(ValueError) as error:
        utils_benchmarks.get_covariate(info_ackley, np.array([0.0, 0.0]), 3)

    covariate = utils_benchmarks.get_covariate(info_branin,
                                               np.array([0.0, 0.0]), 2)
    assert (covariate == np.array([0.0, 0.0])).all()

    covariate = utils_benchmarks.get_covariate(info_ackley, np.array([0.0]), 5)
    assert (covariate == np.array([0.0, 0.0, 0.0, 0.0, 0.0])).all()