def test_param_schema_implicit(self): mv = ModuleVersion.create_or_replace_from_spec( { "id_name": "x", "name": "x", "category": "Clean", "parameters": [ { "id_name": "foo", "type": "string", "default": "X" }, { "id_name": "bar", "type": "secret", "secret_logic": { "provider": "oauth", "service": "google" }, }, { "id_name": "baz", "type": "menu", "options": [ { "value": "a", "label": "A" }, "separator", { "value": "c", "label": "C" }, ], "default": "c", }, ], }, source_version_hash="1.0", ) self.assertEqual( repr(mv.param_schema), repr( ParamDType.Dict({ "foo": ParamDType.String(default="X"), # secret is not in param_schema "baz": ParamDType.Enum(choices=frozenset({"a", "c"}), default="c"), })), )
def test_param_schema_implicit(self): mv = ModuleVersion.create_or_replace_from_spec( { 'id_name': 'x', 'name': 'x', 'category': 'Clean', 'parameters': [ { 'id_name': 'foo', 'type': 'string', 'default': 'X' }, { 'id_name': 'bar', 'type': 'secret', 'secret_logic': { 'provider': 'oauth', 'service': 'google' } }, { 'id_name': 'baz', 'type': 'menu', 'options': [ { 'value': 'a', 'label': 'A' }, 'separator', { 'value': 'c', 'label': 'C' }, ], 'default': 'c' }, ] }, source_version_hash='1.0') self.assertEqual( repr(mv.param_schema), repr( ParamDType.Dict({ 'foo': ParamDType.String(default='X'), # secret is not in param_schema 'baz': ParamDType.Enum(choices=frozenset({'a', 'c'}), default='c'), })))