test: add test for paramType property

pull/12808/head
Matthias 2 months ago
parent bea006880b
commit f6e465a0ba

@ -43,6 +43,7 @@ def test_hyperopt_int_parameter():
HyperoptStateContainer.set_state(HyperoptState.OPTIMIZE)
assert len(list(intpar.range)) == 1
assert intpar.param_type == "IntParameter"
def test_hyperopt_real_parameter():
@ -60,6 +61,7 @@ def test_hyperopt_real_parameter():
assert isinstance(fltpar.get_space(""), FloatDistribution)
assert not hasattr(fltpar, "range")
assert fltpar.param_type == "RealParameter"
def test_hyperopt_decimal_parameter():
@ -94,6 +96,7 @@ def test_hyperopt_decimal_parameter():
HyperoptStateContainer.set_state(HyperoptState.OPTIMIZE)
assert len(list(decimalpar.range)) == 1
assert decimalpar.param_type == "DecimalParameter"
def test_hyperopt_categorical_parameter():
@ -133,3 +136,5 @@ def test_hyperopt_categorical_parameter():
HyperoptStateContainer.set_state(HyperoptState.OPTIMIZE)
assert len(list(catpar.range)) == 1
assert len(list(boolpar.range)) == 1
assert boolpar.param_type == "BooleanParameter"
assert catpar.param_type == "CategoricalParameter"

Loading…
Cancel
Save