split_blocks: Fix dimension checks.

Make dimensions checks of the shape, block shape, and halo shape proper `ValueError`s instead of being assertions.
jakirkham · Dec 4, 2016 · bd6c82d · bd6c82d
1 parent e94ca4d
commit bd6c82d
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/kenjutsu/kenjutsu.py b/kenjutsu/kenjutsu.py
@@ -333,13 +333,17 @@ def split_blocks(space_shape, block_shape, block_halo=None):
         ifilter, imap = filter, map
 
     if block_halo is not None:
-        assert (len(space_shape) == len(block_shape) == len(block_halo)), \
-            "The dimensions of `space_shape`, `block_shape`, and " + \
-            "`block_halo` should be the same."
+        if not (len(space_shape) == len(block_shape) == len(block_halo)):
+            raise ValueError(
+                "The dimensions of `space_shape`, `block_shape`, and"
+                " `block_halo` should be the same."
+            )
     else:
-        assert (len(space_shape) == len(block_shape)), \
-            "The dimensions of `space_shape` and `block_shape` " + \
-            "should be the same."
+        if not (len(space_shape) == len(block_shape)):
+            raise ValueError(
+               "The dimensions of `space_shape` and `block_shape` should be"
+               " the same."
+            )
 
         block_halo = tuple()
         for i in irange(len(space_shape)):