def it_validates_type():
test_s = s(s.is_type(TestType))
test_s.validate(TestType())
with zest.raises(SchemaValidationFailed):
test_s.validate("a str")
with zest.raises(SchemaValidationFailed):
test_s.validate(1.0)
class RFTrainV2Params(Params):
defaults = Munch(
n_subsample=None,
n_estimators=10,
min_samples_leaf=50,
max_depth=None,
max_features="auto",
max_leaf_nodes=None,
)
schema = s(
s.is_kws_r(
n_subsample=s.is_int(noneable=True),
n_estimators=s.is_int(),
min_samples_leaf=s.is_int(),
max_depth=s.is_int(noneable=True),
max_features=s.is_type(object),
max_leaf_nodes=s.is_int(noneable=True),
))
def it_validates_noneable():
test_s = s(s.is_type(TestType))
with zest.raises(SchemaValidationFailed):
test_s.validate(None)
test_s = s(s.is_type(TestType, noneable=True))
test_s.validate(None)
