handle delr/delc and ensure size is correct

modflowpy · Nov 20, 2023 · d81c565 · d81c565
1 parent 597ab8f
commit d81c565
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/flopy/utils/gridutil.py b/flopy/utils/gridutil.py
@@ -94,6 +94,13 @@ def get_disu_kwargs(
     def get_nn(k, i, j):
         return k * nrow * ncol + i * ncol + j
 
+    if not isinstance(delr, np.ndarray):
+        delr = np.array(delr)
+    if not isinstance(delc, np.ndarray):
+        delc = np.array(delc)
+    assert delr.shape == (ncol,)
+    assert delc.shape == (nrow,)
+
     nodes = nlay * nrow * ncol
     iac = np.zeros((nodes), dtype=int)
     ja = []