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'),
}),
})))
