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)