def test_zonbud_active_areas_zone_zero(rtol=1e-2):
try:
import pandas as pd
except ImportError:
'Pandas is not available'
# Read ZoneBudget executable output and reformat
zbud_f = os.path.join(loadpth, 'zonef_mlt_active_zone_0.2.csv')
zbud = pd.read_csv(zbud_f)
zbud.columns = [c.strip() for c in zbud.columns]
zbud.columns = ['_'.join(c.split()) for c in zbud.columns]
zbud.index = pd.Index(['ZONE_{}'.format(z) for z in zbud.ZONE.values],
name='name')
cols = [c for c in zbud.columns if 'ZONE_' in c]
zbud = zbud[cols]
# Run ZoneBudget utility and reformat output
zon_f = os.path.join(loadpth, 'zonef_mlt_active_zone_0.zbr')
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
fpbud = zb.get_dataframes().reset_index()
fpbud = fpbud[['name'] + [c for c in fpbud.columns if 'ZONE' in c]]
fpbud = fpbud.set_index('name').T
fpbud = fpbud[[c for c in fpbud.columns if 'ZONE' in c]]
fpbud = fpbud.loc[['ZONE_{}'.format(z) for z in range(1, 4)]]
# Test for equality
allclose = np.allclose(zbud, fpbud, rtol)
s = 'Zonebudget arrays do not match.'
assert allclose, s
return
def test_get_budget():
zon = read_zbarray(zon_f)
aliases = {1: 'Trey', 2: 'Mike', 4: 'Wilson', 0: 'Carini'}
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 0), aliases=aliases)
zb.get_budget(names='FROM_CONSTANT_HEAD', zones=1)
zb.get_budget(names=['FROM_CONSTANT_HEAD'], zones=[1, 2])
zb.get_budget(net=True)
return
def test_get_budget():
zon = read_zbarray(zon_f)
aliases = {1: 'Trey', 2: 'Mike', 4: 'Wilson', 0: 'Carini'}
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 0), aliases=aliases)
zb.get_budget(names='FROM_CONSTANT_HEAD', zones=1)
zb.get_budget(names=['FROM_CONSTANT_HEAD'], zones=[1, 2])
zb.get_budget(net=True)
return
def test_zonbud_copy():
"""
t039 Test zonbud copy
"""
zon = read_zbarray(zon_f)
cfd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cfd2 = cfd.copy()
assert cfd is not cfd2, 'Copied object is a shallow copy.'
return
def test_zonbud_copy():
"""
t039 Test zonbud copy
"""
zon = read_zbarray(zon_f)
cfd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cfd2 = cfd.copy()
assert cfd is not cfd2, 'Copied object is a shallow copy.'
return
def test_zonbud_aliases():
"""
t039 Test zonbud aliases
"""
zon = read_zbarray(zon_f)
aliases = {1: 'Trey', 2: 'Mike', 4: 'Wilson', 0: 'Carini'}
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096), aliases=aliases, verbose=True)
bud = zb.get_budget()
assert bud[bud['name'] == 'FROM_Mike'].shape[0] > 0, 'No records returned.'
return
def test_zonbud_aliases():
"""
t039 Test zonbud aliases
"""
zon = read_zbarray(zon_f)
aliases = {1: 'Trey', 2: 'Mike', 4: 'Wilson', 0: 'Carini'}
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096), aliases=aliases, verbose=True)
bud = zb.get_budget()
assert bud[bud['name'] == 'FROM_Mike'].shape[0] > 0, 'No records returned.'
return
def test_zonbud_math():
"""
t039 Test zonbud math methods
"""
cbc_f = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
cmd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cmd / 35.3147
cmd * 12.
return
def test_zonbud_copy():
"""
t039 Test zonbud copy
"""
cbc_f = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
cfd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cfd2 = cfd.copy()
assert cfd is not cfd2, 'Copied object is a shallow copy.'
return
def test_zonbud_readwrite_zbarray():
"""
t039 Test zonbud read write
"""
x = np.random.randint(100, 200, size=(5, 150, 200))
write_zbarray(os.path.join(outpth, 'randint'), x)
write_zbarray(os.path.join(outpth, 'randint'), x, fmtin=35, iprn=2)
z = read_zbarray(os.path.join(outpth, 'randint'))
assert np.array_equal(x, z), 'Input and output arrays do not match.'
return
def test_zonbud_readwrite_zbarray():
"""
t039 Test zonbud read write
"""
x = np.random.randint(100, 200, size=(5, 150, 200))
write_zbarray(os.path.join(outpth, 'randint'), x)
write_zbarray(os.path.join(outpth, 'randint'), x, fmtin=35, iprn=2)
z = read_zbarray(os.path.join(outpth, 'randint'))
assert np.array_equal(x, z), 'Input and output arrays do not match.'
return
def test_zonbud_get_record_names():
"""
t039 Test zonbud get_record_names method
"""
cbc_f = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 0))
recnames = zb.get_record_names()
assert len(recnames) > 0, 'No record names returned.'
return
def test_zonbud_get_record_names():
"""
t039 Test zonbud get_record_names method
"""
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 0))
recnames = zb.get_record_names()
assert len(recnames) > 0, 'No record names returned.'
recnames = zb.get_record_names(stripped=True)
assert len(recnames) > 0, 'No record names returned.'
return
def test_zonbud_get_record_names():
"""
t039 Test zonbud get_record_names method
"""
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 0))
recnames = zb.get_record_names()
assert len(recnames) > 0, 'No record names returned.'
recnames = zb.get_record_names(stripped=True)
assert len(recnames) > 0, 'No record names returned.'
return
def test_dataframes():
try:
import pandas
zon = read_zbarray(zon_f)
cmd = ZoneBudget(cbc_f, zon, totim=1095.)
df = cmd.get_dataframes()
assert len(df) > 0, 'Output DataFrames empty.'
except ImportError as e:
print('Skipping DataFrames test, pandas not installed.')
print(e)
return
def test_zonbud_math():
"""
t039 Test zonbud math methods
"""
zon = read_zbarray(zon_f)
cmd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cmd / 35.3147
cmd * 12.
cmd + 1e6
cmd - 1e6
return
def test_dataframes():
try:
import pandas
zon = read_zbarray(zon_f)
cmd = ZoneBudget(cbc_f, zon, totim=1095.)
df = cmd.get_dataframes()
assert len(df) > 0, 'Output DataFrames empty.'
except ImportError as e:
print('Skipping DataFrames test, pandas not installed.')
print(e)
return
def test_zonbud_math():
"""
t039 Test zonbud math methods
"""
zon = read_zbarray(zon_f)
cmd = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096))
cmd / 35.3147
cmd * 12.
cmd + 1e6
cmd - 1e6
return
def test_zonbud_to_csv():
"""
t039 Test zonbud export to csv file method
"""
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, kstpkper=[(0, 1094), (0, 1096)])
f_out = os.path.join(outpth, 'test.csv')
zb.to_csv(f_out)
with open(f_out, 'r') as f:
lines = f.readlines()
assert len(lines) > 0, 'No data written to csv file.'
return
def test_zonbud_aliases():
"""
t039 Test zonbud aliases
"""
cbc_f = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
aliases = {1: 'Trey', 2: 'Mike', 4: 'Wilson', 0: 'Carini'}
zb = ZoneBudget(cbc_f, zon, kstpkper=(0, 1096), aliases=aliases)
bud = zb.get_budget()
m = bud['name'] == 'Mike_IN'
assert bud[m].shape[0] > 0, 'No records returned.'
return
def test_zonbud_to_csv():
"""
t039 Test zonbud export to csv file method
"""
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, kstpkper=[(0, 1094), (0, 1096)])
f_out = os.path.join(outpth, 'test.csv')
zb.to_csv(f_out)
with open(f_out, 'r') as f:
lines = f.readlines()
assert len(lines) > 0, 'No data written to csv file.'
return
def test_zonbud_to_csv():
"""
t039 Test zonbud export to csv file method
"""
cbc_f = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
zb = ZoneBudget(cbc_f, zon, kstpkper=[(0, 1094), (0, 1096)])
zb.to_csv(os.path.join(outpth, 'test.csv'))
with open(os.path.join(outpth, 'test.csv'), 'r') as f:
lines = f.readlines()
assert len(lines) > 0, 'No data written to csv file.'
return
def test_compare2zonebudget(rtol=1e-2):
"""
t039 Compare output from zonbud.exe to the budget calculated by zonbud
utility using the multilayer transient freyberg model.
"""
zonebudget_recarray = read_zonebudget_file(os.path.join(loadpth,
'zonebudget_mlt.csv'))
zon = read_zbarray(os.path.join(loadpth, 'zonef_mlt'))
cbc_fname = os.path.join(loadpth, 'freyberg_mlt', 'freyberg.gitcbc')
zb = ZoneBudget(cbc_fname, zon, verbose=False)
zbutil_recarray = zb.get_budget()
times = np.unique(zonebudget_recarray['totim'])
print(times)
zonenames = [n for n in zonebudget_recarray.dtype.names if 'ZONE' in n]
for time in times:
print('Time:', time)
zb_arr = zonebudget_recarray[zonebudget_recarray['totim'] == time]
zbu_arr = zbutil_recarray[zbutil_recarray['totim'] == time]
for name in zbu_arr['name']:
r1 = np.where((zb_arr['name'] == name))
r2 = np.where((zbu_arr['name'] == name))
if r1[0].shape[0] < 1 or r2[0].shape[0] < 1:
continue
if r1[0].shape[0] != r2[0].shape[0]:
continue
a1 = np.array([v for v in zb_arr[zonenames][r1[0]][0]])
a2 = np.array([v for v in zbu_arr[zonenames][r2[0]][0]])
allclose = np.allclose(a1, a2, rtol)
mxdiff = np.abs(a1 - a2).max()
idxloc = np.argmax(np.abs(a1 - a2))
txt = '{} - Max: {} a1: {} a2: {}'.format(name, mxdiff,
a1[idxloc],
a2[idxloc])
print(txt)
s = 'Zonebudget arrays do not match at time {0} ({1}): {2}.' \
.format(time, name, mxdiff)
assert allclose, s
return
def test_compare2zonebudget(rtol=1e-2):
"""
t039 Compare output from zonbud.exe to the budget calculated by zonbud
utility using the multilayer transient freyberg model.
"""
zba = read_zonebudget_file(zbud_f)
zonenames = [n for n in zba.dtype.names if 'ZONE' in n]
times = np.unique(zba['totim'])
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, totim=times, verbose=False)
fpa = zb.get_budget()
for time in times:
zb_arr = zba[zba['totim'] == time]
fp_arr = fpa[fpa['totim'] == time]
for name in fp_arr['name']:
r1 = np.where((zb_arr['name'] == name))
r2 = np.where((fp_arr['name'] == name))
if r1[0].shape[0] < 1 or r2[0].shape[0] < 1:
continue
if r1[0].shape[0] != r2[0].shape[0]:
continue
a1 = np.array([v for v in zb_arr[zonenames][r1[0]][0]])
a2 = np.array([v for v in fp_arr[zonenames][r2[0]][0]])
allclose = np.allclose(a1, a2, rtol)
mxdiff = np.abs(a1 - a2).max()
idxloc = np.argmax(np.abs(a1 - a2))
# txt = '{}: {} - Max: {} a1: {} a2: {}'.format(time,
# name,
# mxdiff,
# a1[idxloc],
# a2[idxloc])
# print(txt)
s = 'Zonebudget arrays do not match at time {0} ({1}): {2}.' \
.format(time, name, mxdiff)
assert allclose, s
return
def test_compare2zonebudget(rtol=1e-2):
"""
t039 Compare output from zonbud.exe to the budget calculated by zonbud
utility using the multilayer transient freyberg model.
"""
zba = read_zonebudget_file(zbud_f)
zonenames = [n for n in zba.dtype.names if 'ZONE' in n]
times = np.unique(zba['totim'])
zon = read_zbarray(zon_f)
zb = ZoneBudget(cbc_f, zon, totim=times, verbose=False)
fpa = zb.get_budget()
for time in times:
zb_arr = zba[zba['totim'] == time]
fp_arr = fpa[fpa['totim'] == time]
for name in fp_arr['name']:
r1 = np.where((zb_arr['name'] == name))
r2 = np.where((fp_arr['name'] == name))
if r1[0].shape[0] < 1 or r2[0].shape[0] < 1:
continue
if r1[0].shape[0] != r2[0].shape[0]:
continue
a1 = np.array([v for v in zb_arr[zonenames][r1[0]][0]])
a2 = np.array([v for v in fp_arr[zonenames][r2[0]][0]])
allclose = np.allclose(a1, a2, rtol)
mxdiff = np.abs(a1 - a2).max()
idxloc = np.argmax(np.abs(a1 - a2))
# txt = '{}: {} - Max: {} a1: {} a2: {}'.format(time,
# name,
# mxdiff,
# a1[idxloc],
# a2[idxloc])
# print(txt)
s = 'Zonebudget arrays do not match at time {0} ({1}): {2}.' \
.format(time, name, mxdiff)
assert allclose, s
return
import os import sys import platform import numpy as np import matplotlib as mpl import matplotlib.pyplot as plt import pandas as pd import flopy from flopy.utils import read_zbarray # Set path to example datafiles cbc_f = sys.argv[1] zon = read_zbarray(sys.argv[2]) zb = flopy.utils.ZoneBudget(cbc_f, zon, kstpkper=None) zb.to_csv(sys.argv[3])
def test_get_model_shape():
ZoneBudget(cbc_f, read_zbarray(zon_f), kstpkper=(0, 0), verbose=True).get_model_shape()
return
def test_get_model_shape():
ZoneBudget(cbc_f, read_zbarray(zon_f), kstpkper=(0, 0),
verbose=True).get_model_shape()
return
def test_zonebudget_output_to_netcdf():
from flopy.utils import HeadFile, ZoneBudgetOutput
from flopy.modflow import Modflow
from flopy.mf6 import MFSimulation
from flopy.export.utils import output_helper
model_ws = os.path.join("..", "examples", "data",
"freyberg_multilayer_transient")
zb_ws = os.path.join("..", "examples", "data", "zonbud_examples")
hds = "freyberg.hds"
nam = "freyberg.nam"
zon = "zonef_mlt.zbr"
hds = HeadFile(os.path.join(model_ws, hds))
ml = Modflow.load(nam, model_ws=model_ws)
zone_array = read_zbarray(os.path.join(zb_ws, zon))
# test with standard zonebudget output
zbout = "freyberg_mlt.txt"
ncf_name = zbout + ".nc"
zb = ZoneBudgetOutput(os.path.join(zb_ws, zbout), ml.dis, zone_array)
vdf = zb.volumetric_flux()
netobj = zb.dataframe_to_netcdf_fmt(vdf, flux=False)
export_dict = {"hds": hds, "zonebud": netobj}
output_helper(os.path.join(outpth, ncf_name), ml, export_dict)
# test with zonebudget csv 1 output
zbout = "freyberg_mlt.1.csv"
ncf_name = zbout + ".nc"
zb = ZoneBudgetOutput(os.path.join(zb_ws, zbout), ml.dis, zone_array)
netobj = zb.dataframe_to_netcdf_fmt(zb.dataframe)
export_dict = {"hds": hds, "zonebud": netobj}
output_helper(os.path.join(outpth, ncf_name), ml, export_dict)
# test with zonebudget csv 2 output
zbout = "freyberg_mlt.2.csv"
ncf_name = zbout + ".nc"
zb = ZoneBudgetOutput(os.path.join(zb_ws, zbout), ml.dis, zone_array)
vdf = zb.volumetric_flux(extrapolate_kper=True)
netobj = zb.dataframe_to_netcdf_fmt(vdf, flux=False)
export_dict = {"hds": hds, "zonebud": netobj}
output_helper(os.path.join(outpth, ncf_name), ml, export_dict)
# test built in export function
zbout = "freyberg_mlt.2.csv"
ncf_name = zbout + ".bi1" + ".nc"
zb = ZoneBudgetOutput(os.path.join(zb_ws, zbout), ml.dis, zone_array)
zb.export(os.path.join(outpth, ncf_name), ml)
# test built in export function with NetCdf output object
zbout = "freyberg_mlt.2.csv"
ncf_name = zbout + ".bi2" + ".nc"
zb = ZoneBudgetOutput(os.path.join(zb_ws, zbout), ml.dis, zone_array)
export_dict = {"hds": hds}
ncfobj = output_helper(os.path.join(outpth, ncf_name), ml, export_dict)
zb.export(ncfobj, ml)
# test with modflow6/zonebudget6
sim_ws = os.path.join("..", "examples", 'data', 'mf6',
'test005_advgw_tidal')
hds = "advgw_tidal.hds"
nam = "mfsim"
zon = "zonebudget6.csv"
ncf_name = zon + ".nc"
zone_array = np.ones((3, 15, 10), dtype=int)
zone_array = np.add.accumulate(zone_array, axis=0)
sim = MFSimulation.load(nam, sim_ws=sim_ws, exe_name='mf6')
sim.set_sim_path(outpth)
sim.write_simulation()
sim.run_simulation()
hds = HeadFile(os.path.join(outpth, hds))
ml = sim.get_model("gwf_1")
zb = ZoneBudgetOutput(os.path.join(zb_ws, zon), sim.tdis, zone_array)
vdf = zb.volumetric_flux()
netobj = zb.dataframe_to_netcdf_fmt(vdf, flux=False)
export_dict = {"hds": hds, "zbud": netobj}
output_helper(os.path.join(outpth, ncf_name), ml, export_dict)
