def test_override_array_param(self):
init = self.sk_pca.hyperparam_schema('copy')
expected = {'type': 'array',
'minItems': 1,
'maxItems': 20,
'items': {'type': 'integer'}}
foo = self.sk_pca.customize_schema(
copy=schemas.Array(minItems=1, maxItems=20, items=schemas.Int()))
self.assertEqual(foo.hyperparam_schema('copy'), expected)
lale.type_checking.validate_is_schema(foo._schemas)
self.assertEqual(self.sk_pca.hyperparam_schema('copy'), init)
def test_override_any_param(self):
init = self.ll_pca.hyperparam_schema('iterated_power')
expected = {'anyOf': [
{'type': 'integer'},
{'enum': ['auto', 'full']}],
'default': 'auto'}
foo = self.ll_pca.customize_schema(
iterated_power=schemas.AnyOf([schemas.Int(),
schemas.Enum(['auto', 'full'])], default='auto'))
self.assertEqual(foo.hyperparam_schema('iterated_power'), expected)
lale.type_checking.validate_is_schema(foo._schemas)
self.assertEqual(self.ll_pca.hyperparam_schema('iterated_power'), init)
def test_override_array_param(self):
init = self.sk_pca.hyperparam_schema("copy")
expected = {
"type": "array",
"minItems": 1,
"maxItems": 20,
"items": {"type": "integer"},
}
foo = self.sk_pca.customize_schema(
copy=schemas.Array(minItems=1, maxItems=20, items=schemas.Int())
)
self.assertEqual(foo.hyperparam_schema("copy"), expected)
lale.type_checking.validate_is_schema(foo._schemas)
self.assertEqual(self.sk_pca.hyperparam_schema("copy"), init)
def test_override_any_param(self):
init = self.ll_pca.hyperparam_schema("iterated_power")
expected = {
"anyOf": [{"type": "integer"}, {"enum": ["auto", "full"]}],
"default": "auto",
}
foo = self.ll_pca.customize_schema(
iterated_power=schemas.AnyOf(
[schemas.Int(), schemas.Enum(["auto", "full"])], default="auto"
)
)
self.assertEqual(foo.hyperparam_schema("iterated_power"), expected)
lale.type_checking.validate_is_schema(foo._schemas)
self.assertEqual(self.ll_pca.hyperparam_schema("iterated_power"), init)
def test_override_int_param(self):
init = self.ll_pca.hyperparam_schema('iterated_power')
expected = {
'default': 1,
'type': 'integer',
'minimum': -10,
'maximum': 10,
'exclusiveMaximum': True,
'exclusiveMinimum': False
}
foo = self.ll_pca.customize_schema(iterated_power=schemas.Int(
default=1, min=-10, max=10, exclusiveMax=True, exclusiveMin=False))
self.assertEqual(foo.hyperparam_schema('iterated_power'), expected)
helpers.validate_is_schema(foo._schemas)
self.assertEqual(self.ll_pca.hyperparam_schema('iterated_power'), init)
def test_override_int_param(self):
init = self.ll_pca.hyperparam_schema("iterated_power")
expected = {
"default": 1,
"type": "integer",
"minimum": -10,
"maximum": 10,
"exclusiveMaximum": True,
"exclusiveMinimum": False,
}
foo = self.ll_pca.customize_schema(iterated_power=schemas.Int(
default=1, min=-10, max=10, exclusiveMax=True, exclusiveMin=False))
self.assertEqual(foo.hyperparam_schema("iterated_power"), expected)
lale.type_checking.validate_is_schema(foo._schemas)
self.assertEqual(self.ll_pca.hyperparam_schema("iterated_power"), init)
