Merge pull request #16 from bionumpy/dev

Dev
bionumpy · Mar 6, 2024 · 029a870 · 029a870
2 parents c3b9eef + f5e075f
commit 029a870
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/npstructures/raggedarray/__init__.py b/npstructures/raggedarray/__init__.py
@@ -96,6 +96,8 @@ def __init__(self, data: npt.ArrayLike, shape: ShapeLike = None, dtype: npt.DTyp
             shape = shape
         else:
             shape = RaggedShape.asshape(shape)
+            if not shape.size == len(data) and safe_mode:
+                raise ValueError(f"The total size of provided shape {shape.size} does not match the size of the data array: {len(data)}")
         if not hasattr(data, "__array_ufunc__"):
             data = np.asanyarray(data, dtype=dtype)
         super().__init__(data, shape)

diff --git a/tests/test_raggedarray_usage.py b/tests/test_raggedarray_usage.py
@@ -0,0 +1,11 @@
+import pytest
+
+from npstructures import RaggedArray
+
+# this works
+
+def test_assert_raises_value_error():
+    with pytest.raises(ValueError):
+        x = RaggedArray([0, 1, 2, 3], [4, 5, 6])
+
+