def test_param_schema_explicit(): spec = load_spec( dict( 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"}, }, } }, ) ) assert spec.param_schema == ParamSchema.Dict( { "id_name": ParamSchema.Dict( {"x": ParamSchema.Integer(), "y": ParamSchema.String(default="X")} ) } )
def test_clean_normal_dict(self): schema = ParamSchema.Dict({ "str": ParamSchema.String(), "int": ParamSchema.Integer() }) value = {"str": "foo", "int": 3} expected = dict(value) # no-op result = self._call_clean_value(schema, value) self.assertEqual(result, expected)
def test_clean_normal_dict(self): input_shape = TableMetadata(3, [Column("A", ColumnType.Number())]) schema = ParamSchema.Dict({ "str": ParamSchema.String(), "int": ParamSchema.Integer() }) value = {"str": "foo", "int": 3} expected = dict(value) # no-op result = clean_value(schema, value, input_shape) self.assertEqual(result, expected)
def test_validate_ok(self): S.Dict({"foo": S.String(default="FOO"), "bar": S.Integer(default=3)}).validate( {"foo": "FOO", "bar": 3} )
def test_default(self): assert S.Dict( {"foo": S.String(default="FOO"), "bar": S.Integer(default=3)} ).default == {"foo": "FOO", "bar": 3}