def test_debug_level():
from hpobench.util.container_utils import enable_container_debug, disable_container_debug
import os
enable_container_debug()
assert os.environ['HPOBENCH_DEBUG'] == 'true'
disable_container_debug()
assert os.environ['HPOBENCH_DEBUG'] == 'false'
import pytest
from hpobench.container.benchmarks.ml.pybnn import BNNOnToyFunction, BNNOnBostonHousing, BNNOnProteinStructure, \
BNNOnYearPrediction
import logging
logging.basicConfig(level=logging.DEBUG)
from hpobench.util.container_utils import enable_container_debug
enable_container_debug()
def test_bnn_init():
benchmark = BNNOnToyFunction(rng=1)
fs = benchmark.get_fidelity_space(seed=0)
fidelity = fs.sample_configuration().get_dictionary()
assert fidelity['budget'] == 5714
meta = benchmark.get_meta_information()
assert meta is not None
cs = benchmark.get_configuration_space(seed=0)
config = cs.sample_configuration().get_dictionary()
assert config['l_rate'] == pytest.approx(0.0037, abs=0.001)
assert config['burn_in'] == pytest.approx(0.43905, abs=0.001)
assert config['n_units_1'] == 104
assert config['n_units_2'] == 68
assert config['mdecay'] == pytest.approx(0.6027, abs=0.001)
result = benchmark.objective_function(configuration=config,
def enable_debug():
enable_container_debug()
yield
disable_container_debug()