diff --git a/autotest/t061_test_gridgen.py b/autotest/t061_test_gridgen.py index 27ff295337..b2b1c88f84 100644 --- a/autotest/t061_test_gridgen.py +++ b/autotest/t061_test_gridgen.py @@ -114,6 +114,18 @@ def test_gridgen(): mu = flopy.modflow.Modflow(version='mfusg', structured=False) disu = g.get_disu(mu) + # test writing a modflow 6 disu package + fname = os.path.join(cpth, 'mymf6model.disu') + g6.to_disu6(fname) + assert os.path.isfile(fname), \ + 'MF6 disu file not created: {}'.format(fname) + + # test writing a modflow 6 disv package + fname = os.path.join(cpth, 'mymf6model.disv') + g6.to_disv6(fname) + assert os.path.isfile(fname), \ + 'MF6 disv file not created: {}'.format(fname) + return diff --git a/flopy/utils/gridgen.py b/flopy/utils/gridgen.py index fc7521a6aa..2d2c986c33 100644 --- a/flopy/utils/gridgen.py +++ b/flopy/utils/gridgen.py @@ -1287,7 +1287,7 @@ def to_disu6(self, fname, writevertices=True): """ - gridprops = self.get_gridprops() + gridprops = self.get_gridprops_disu6() f = open(fname, 'w') # opts @@ -1338,7 +1338,7 @@ def to_disu6(self, fname, writevertices=True): # celldata -- not optimized for redundant vertices yet f.write('BEGIN CELL2D\n') iv = 1 - for n in range(nodes): + for n in range(gridprops['nodes']): xc, yc = self.get_center(n) s = ' {} {} {} {} {} {} {} {}\n'.format(n + 1, xc, yc, 4, iv, iv + 1, iv + 2,