def test_clean_normal_dict(self): input_shape = TableShape(3, [Column("A", ColumnType.NUMBER())]) schema = ParamDType.Dict( {"str": ParamDType.String(), "int": ParamDType.Integer()} ) value = {"str": "foo", "int": 3} expected = dict(value) # no-op result = clean_value(schema, value, input_shape) self.assertEqual(result, expected)
def test_clean_normal_dict(self): context = RenderContext(None, None, None, None, None) schema = ParamDType.Dict({ 'str': ParamDType.String(), 'int': ParamDType.Integer(), }) value = {'str': 'foo', 'int': 3} expected = dict(value) # no-op result = clean_value(schema, value, context) self.assertEqual(result, expected)
def test_clean_normal_dict(self): input_shape = TableShape(3, [Column('A', ColumnType.NUMBER())]) schema = ParamDType.Dict({ 'str': ParamDType.String(), 'int': ParamDType.Integer(), }) value = {'str': 'foo', 'int': 3} expected = dict(value) # no-op result = clean_value(schema, value, input_shape) self.assertEqual(result, expected)
def test_param_schema_explicit(self): mv = ModuleVersion.create_or_replace_from_spec( { "id_name": "x", "name": "x", "category": "Clean", "parameters": [{ "id_name": "whee", "type": "custom" }], "param_schema": { "id_name": { "type": "dict", "properties": { "x": { "type": "integer" }, "y": { "type": "string", "default": "X" }, }, } }, }, source_version_hash="1.0", ) self.assertEqual( repr(mv.param_schema), repr( ParamDType.Dict({ "id_name": ParamDType.Dict({ "x": ParamDType.Integer(), "y": ParamDType.String(default="X"), }) })), )
def test_param_schema_explicit(self): mv = ModuleVersion.create_or_replace_from_spec({ 'id_name': 'x', 'name': 'x', 'category': 'Clean', 'parameters': [ {'id_name': 'whee', 'type': 'custom'} ], 'param_schema': { 'id_name': { 'type': 'dict', 'properties': { 'x': {'type': 'integer'}, 'y': {'type': 'string', 'default': 'X'}, }, }, }, }, source_version_hash='1.0') self.assertEqual(repr(mv.param_schema), repr(ParamDType.Dict({ 'id_name': ParamDType.Dict({ 'x': ParamDType.Integer(), 'y': ParamDType.String(default='X'), }), })))