class MySchema(ArgSchema):
a = NumpyArray(dtype='float',
description='Test input array schema',
validate=validate.Shape((2, 2)))
b = NumpyArray(dtype='float',
description='Test array',
validate=validate.Shape((2, None)))
def test_shape_call(validation_shape, input_array):
validator = validate.Shape(validation_shape)
assert (validator(input_array))
def test_shape_init(input_shape, expected):
validator = validate.Shape(input_shape)
assert (validator.shape == expected)
def test_shape_call_invalid(validation_shape, input_array):
validator = validate.Shape(validation_shape)
with pytest.raises(mm.ValidationError):
validator(input_array)
def test_shape_init_invalid(invalid_shape):
with pytest.raises(ValueError):
validate.Shape(invalid_shape)
