Let sparse work in both Chd and Oc

Author: deltamarnix

docs/examples/quickstart.py

@@ -29,7 +29,7 @@
 # check CHD
 assert chd.data["head"][0, 0].item() == 1.0
 assert chd.data["head"][0, 99].item() == 0.0
-assert np.allclose(chd.data["head"][:, 1:99], np.full(98, 1e30))
+assert np.allclose(chd.data["head"][:, 1:99].data.todense(), np.full(98, 1e30))
 
 # TODO: xarray index aliasing nlay/ncol/nrow to k/i/j?
 # assert chd.data["head"].loc(dict(k=0, i=0, j=0)) == 1.

flopy4/mf6/converters.py

@@ -52,7 +52,7 @@ def _get_nn(cellid):
                 kper = 0
             match len(shape):
                 case 1:
-                    a[(kper)] = period
+                    a[(kper,)] = period
                 case _:
                     for cellid, v in period.items():
                         nn = _get_nn(cellid)
@@ -62,7 +62,7 @@ def _get_nn(cellid):
     else:
         for cellid, v in value.items():
             nn = _get_nn(cellid)
-            a[(nn)] = v
+            a[(nn,)] = v
 
     coords = np.array(list(map(list, zip(*a.keys()))))
     return sparse.COO(