Check that array shape is not larger than MAX_DIM

Author: FrancescAlted

src/blosc2/__init__.py

@@ -109,6 +109,7 @@ class Tuner(Enum):
     GLOBAL_REGISTERED_CODECS_STOP,
     MAX_BLOCKSIZE,
     MAX_BUFFERSIZE,
+    MAX_DIM,
     MAX_OVERHEAD,
     MAX_TYPESIZE,
     MIN_HEADER_LENGTH,

src/blosc2/blosc2_ext.pyx

@@ -542,6 +542,7 @@ MAX_TYPESIZE = BLOSC2_MAXTYPESIZE
 MAX_BUFFERSIZE = BLOSC2_MAX_BUFFERSIZE
 MAX_BLOCKSIZE = BLOSC2_MAXBLOCKSIZE
 MAX_OVERHEAD = BLOSC2_MAX_OVERHEAD
+MAX_DIM = B2ND_MAX_DIM
 VERSION_STRING = (<char*>BLOSC2_VERSION_STRING).decode("utf-8")
 VERSION_DATE = (<char*>BLOSC2_VERSION_DATE).decode("utf-8")
 MIN_HEADER_LENGTH = BLOSC_MIN_HEADER_LENGTH

src/blosc2/ndarray.py

@@ -2914,6 +2914,8 @@ def _check_shape(shape):
         shape = (shape,)
     elif not isinstance(shape, tuple | list):
         raise TypeError("shape should be a tuple or a list!")
+    if len(shape) > blosc2.MAX_DIM:
+        raise ValueError(f"shape length {len(shape)} is too large (>{blosc2.MAX_DIM})!")
     return shape
 
 

tests/ndarray/test_zeros.py

@@ -166,3 +166,9 @@ def test_shape_empty():
     assert a.dtype == np.int32
     b = np.zeros((), dtype=np.int32)
     np.testing.assert_equal(a[()], b)
+
+
+def test_shape_max_dims():
+    # Test that the shape cannot exceed the maximum number of dimensions
+    with pytest.raises(ValueError):
+        a = blosc2.zeros((1,) * (blosc2.MAX_DIM + 1), dtype=np.int32)