Fix(unstructured grid capabilities): fix array masking for CVFD grids…

… and allow ncpl to be specified as a list (#631)

* Fix(UnstructuredGrid): fix for ncpl provided as a list or numpy array

* Fix(plot_array): fix for CVFD masked_values when plotting arrays

flopy/discretization/unstructuredgrid.py

@@ -51,7 +51,7 @@ def __init__(self, vertices=None, iverts=None, xcenters=None, ycenters=None,
             else:
                 msg = ('Length of iverts must equal ncpl.sum '
                        '({} {})'.format(len(iverts), ncpl))
-                assert len(iverts) == ncpl.sum(), msg
+                assert len(iverts) == np.sum(ncpl), msg
                 assert np.array(self.xcellcenters).shape[0] == self.ncpl
                 assert np.array(self.ycellcenters).shape[0] == self.ncpl
 

flopy/plot/map.py

@@ -170,7 +170,7 @@ def plot_array(self, a, masked_values=None, **kwargs):
 
         else:
             quadmesh = plotutil.plot_cvfd(self.mg._vertices, self.mg._iverts,
-                                          a=a, ax=ax)
+                                          a=plotarray, ax=ax)
 
         # set max and min
         if 'vmin' in kwargs: