def test_categorical_parameter_value_ranges(): cat_param = CategoricalParameter('a') ranges = cat_param.as_tuning_range('some') assert len(ranges.keys()) == 2 assert ranges['Name'] == 'some' assert ranges['Values'] == ['a']
def test_categorical_parameter_list_ranges(): cat_param = CategoricalParameter([1, 10]) ranges = cat_param.as_tuning_range('some') assert len(ranges.keys()) == 2 assert ranges['Name'] == 'some' assert ranges['Values'] == ['1', '10']