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)