From 624c7b3f05c4d99087e9c9723f13d960a4210dd4 Mon Sep 17 00:00:00 2001 From: Christian Langevin Date: Thu, 17 Oct 2019 14:23:27 -0500 Subject: [PATCH] fix(gridgen): fixed writing of mf6 disu package * Added tests for the to_disu6() and to_disv6() Gridgen methods * Closes #678 --- autotest/t061_test_gridgen.py | 12 ++++++++++++ flopy/utils/gridgen.py | 4 ++-- 2 files changed, 14 insertions(+), 2 deletions(-) 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,