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)