def test006_gwf3():
# init paths
test_ex_name = 'test006_gwf3'
model_name = 'gwf_1'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder,
'flow_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder, 'flow_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder, 'flow_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder, 'flow_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name, 'mf6', exe_name, pth, verify_data=True)
model = sim.get_model()
disu = model.get_package('disu')
# test switching disu array to internal array
disu.ja = disu.ja.array
# test writing hwva and cl12 arrays out to different locations
disu.hwva = {
'filename': 'flow.disu.hwva_new.dat',
'factor': 1.0,
'data': disu.hwva.array
}
disu.cl12 = {
'filename': 'flow.disu.cl12_new.dat',
'factor': 1.0,
'data': disu.cl12.array
}
# make temp folder to save simulation
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.set_all_data_external()
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'flow.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid),
np.array(budget_fjf))
# change some settings
model = sim.get_model(model_name)
hk = model.get_package('npf').k
hk_data = hk.get_data()
hk_data[2] = 3.5
hk.set_data(hk_data)
ex_happened = False
try:
hk.make_layered()
except:
ex_happened = True
assert (ex_happened)
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='auto')
budget_fjf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid),
np.array(budget_fjf))
# confirm that files did move
save_folder = os.path.join(run_folder, 'temp_two')
sim.simulation_data.mfpath.set_sim_path(save_folder)
# write with "copy_external_files" turned off so external files do not get copied to new location
sim.write_simulation(
ext_file_action=flopy.mf6.mfbase.ExtFileAction.copy_none)
# store strt in an external binary file
model = sim.get_model()
ic = model.get_package('ic')
ic.strt.store_as_external_file('initial_heads.bin', binary=True)
strt_data = ic.strt.array
# update packages
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid),
np.array(budget_fjf))
# confirm that files did not move
assert not os.path.isfile(os.path.join(save_folder,
'flow.disu.ja.dat'))
assert not os.path.isfile(
os.path.join(save_folder, 'flow.disu.iac.dat'))
assert not os.path.isfile(
os.path.join(save_folder, 'flow.disu.cl12.dat'))
assert not os.path.isfile(
os.path.join(save_folder, 'flow.disu.area.dat'))
assert not os.path.isfile(
os.path.join(save_folder, 'flow.disu.hwva.dat'))
# confirm external binary file was created
assert os.path.isfile(os.path.join(save_folder, 'initial_heads.bin'))
# clean up
sim.delete_output_files()
return
def test001a_tharmonic():
# init paths
test_ex_name = 'test001a_Tharmonic'
model_name = 'flow15'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder,
'flow15_flow_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder,
'flow15_flow_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder,
'flow15_flow_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder,
'flow15_flow_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name,
'mf6',
exe_name,
pth,
verbosity_level=0,
verify_data=True,
write_headers=False)
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.set_all_data_external()
sim.write_simulation(silent=True)
model = sim.get_model(model_name)
model.export('{}/tharmonic.nc'.format(model.model_ws))
model.export('{}/tharmonic.shp'.format(model.model_ws))
model.dis.botm.export('{}/botm.shp'.format(model.model_ws))
mg = model.modelgrid
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='auto')
budget_obj.list_records()
budget_frf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'flow15_flow.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# change some settings
hk_data = sim.simulation_data.mfdata[(model_name, 'npf', 'griddata', 'k')]
hk_array = hk_data.get_data()
hk_array[0, 0, 1] = 20.0
hk_data.set_data(hk_array)
model = sim.get_model(model_name)
ic = model.get_package('ic')
ic_data = ic.strt
ic_array = ic_data.get_data()
ic_array[0, 0, 0] = 1.0
ic_array[0, 0, 9] = 1.0
ic_data.set_data(ic_array)
get_test = hk_data[0, 0, 0]
assert (get_test == 10.0)
get_test = hk_data.array
assert (array_util.array_comp(
get_test,
[[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]]))
get_test = hk_data[:]
assert (array_util.array_comp(
get_test,
[[[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]]]))
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='auto')
budget_frf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow15_flow.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# clean up
sim.delete_output_files()
return
def test003_gwfs_disv():
# init paths
test_ex_name = 'test003_gwfs_disv'
model_name = 'gwf_1'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder,
'model_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder,
'model_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder,
'model_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder, 'model_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name, 'mf6', exe_name, pth, verify_data=True)
# make temp folder to save simulation
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.simulation_data.max_columns_of_data = 10
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='auto')
budget_fjf_valid = np.array(
budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'model.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')]
assert array_util.array_comp(budget_fjf_valid, budget_frf)
model = sim.get_model(model_name)
if shapefile:
model.export('{}/{}.shp'.format(pth, test_ex_name))
# change some settings
chd_head_left = model.get_package('CHD_LEFT')
chd_left_period = chd_head_left.stress_period_data.get_data(0)
chd_left_period[4][1] = 15.0
chd_head_left.stress_period_data.set_data(chd_left_period, 0)
chd_head_right = model.get_package('CHD_RIGHT')
chd_right_period = chd_head_right.stress_period_data
chd_right_data = chd_right_period.get_data(0)
chd_right_data_slice = chd_right_data[3:10]
chd_right_period.set_data(chd_right_data_slice, 0)
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(
budget_obj.get_data(text='FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'model.hds')
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC',
'FLOW-JA-FACE')]
assert array_util.array_comp(budget_fjf_valid, budget_frf)
# clean up
sim.delete_output_files()
return
def test001a_tharmonic():
# init paths
test_ex_name = "test001a_Tharmonic"
model_name = "flow15"
pth = os.path.join("..", "examples", "data", "mf6", test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, "temp")
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, "expected_output")
expected_head_file_a = os.path.join(expected_output_folder,
"flow15_flow_unch.hds")
expected_head_file_b = os.path.join(expected_output_folder,
"flow15_flow_adj.hds")
expected_cbc_file_a = os.path.join(expected_output_folder,
"flow15_flow_unch.cbc")
expected_cbc_file_b = os.path.join(expected_output_folder,
"flow15_flow_adj.cbc")
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(
model_name,
"mf6",
exe_name,
pth,
verbosity_level=0,
verify_data=True,
write_headers=False,
)
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.set_all_data_external(external_data_folder="data")
sim.write_simulation(silent=True)
# verify external data written to correct location
data_path = os.path.join(run_folder, "data", "flow15.dis_botm.txt")
assert os.path.exists(data_path)
# model export test
model = sim.get_model(model_name)
model.export("{}/tharmonic.nc".format(model.model_ws))
model.export("{}/tharmonic.shp".format(model.model_ws))
model.dis.botm.export("{}/botm.shp".format(model.model_ws))
mg = model.modelgrid
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision="auto")
budget_obj.list_records()
budget_frf_valid = np.array(
budget_obj.get_data(text=" FLOW JA FACE", full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, "flow15_flow.hds")
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, "CBC",
"FLOW-JA-FACE")]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# change some settings
hk_data = sim.simulation_data.mfdata[(model_name, "npf", "griddata", "k")]
hk_array = hk_data.get_data()
hk_array[0, 0, 1] = 20.0
hk_data.set_data(hk_array)
model = sim.get_model(model_name)
ic = model.get_package("ic")
ic_data = ic.strt
ic_array = ic_data.get_data()
ic_array[0, 0, 0] = 1.0
ic_array[0, 0, 9] = 1.0
ic_data.set_data(ic_array)
get_test = hk_data[0, 0, 0]
assert get_test == 10.0
get_test = hk_data.array
assert array_util.array_comp(
get_test,
[[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]],
)
get_test = hk_data[:]
assert array_util.array_comp(
get_test,
[[[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]]],
)
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision="auto")
budget_frf_valid = np.array(
budget_obj.get_data(text=" FLOW JA FACE", full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, "flow15_flow.hds")
assert pymake.compare_heads(None,
None,
files1=head_file,
files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, "CBC",
"FLOW-JA-FACE")]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# clean up
sim.delete_output_files()
return
def test006_gwf3():
# init paths
test_ex_name = 'test006_gwf3'
model_name = 'gwf_1'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder, 'flow_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder, 'flow_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder, 'flow_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder, 'flow_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name, 'mf6', exe_name, pth)
# make temp folder to save simulation
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'flow.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid), np.array(budget_fjf))
# change some settings
model = sim.get_model(model_name)
hk = model.get_package('npf').k
hk_data = hk.get_data()
hk_data[2] = 3.5
hk.set_data(hk_data)
ex_happened = False
try:
hk.make_layered()
except:
ex_happened = True
assert(ex_happened)
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid), np.array(budget_fjf))
# confirm that files did move
save_folder = os.path.join(run_folder, 'temp_two')
sim.simulation_data.mfpath.set_sim_path(save_folder)
# write with "copy_external_files" turned off so external files do not get copied to new location
sim.write_simulation(ext_file_action=flopy.mf6.mfbase.ExtFileAction.copy_none)
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
jaentries = budget_fjf_valid.shape[-1]
budget_fjf_valid.shape = (-1, jaentries)
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_fjf = np.array(sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')])
assert array_util.array_comp(np.array(budget_fjf_valid), np.array(budget_fjf))
# confirm that files did not move
assert not os.path.isfile(os.path.join(save_folder, 'flow.disu.ja.dat'))
assert not os.path.isfile(os.path.join(save_folder, 'flow.disu.iac.dat'))
assert not os.path.isfile(os.path.join(save_folder, 'flow.disu.cl12.dat'))
assert not os.path.isfile(os.path.join(save_folder, 'flow.disu.area.dat'))
assert not os.path.isfile(os.path.join(save_folder, 'flow.disu.hwva.dat'))
# clean up
sim.delete_output_files()
return
def test001a_tharmonic():
# init paths
test_ex_name = 'test001a_Tharmonic'
model_name = 'flow15'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder, 'flow15_flow_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder, 'flow15_flow_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder, 'flow15_flow_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder, 'flow15_flow_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name, 'mf6', exe_name, pth,
verbosity_level=0)
sim.simulation_data.mfpath.set_sim_path(run_folder)
model = sim.get_model(model_name)
model.export('{}/tharmonic.nc'.format(model.model_ws))
model.export('{}/tharmonic.shp'.format(model.model_ws))
model.dis.botm.export('{}/botm.shp'.format(model.model_ws))
# write simulation to new location
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_obj.list_records()
budget_frf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'flow15_flow.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# change some settings
hk_data = sim.simulation_data.mfdata[(model_name, 'npf', 'griddata', 'k')]
hk_array = hk_data.get_data()
hk_array[0,0,1] = 20.0
hk_data.set_data(hk_array)
model = sim.get_model(model_name)
ic = model.get_package('ic')
ic_data = ic.strt
ic_array = ic_data.get_data()
ic_array[0,0,0] = 1.0
ic_array[0,0,9] = 1.0
ic_data.set_data(ic_array)
get_test = hk_data[0,0,0]
assert(get_test == 10.0)
get_test = hk_data.array
assert(array_util.array_comp(get_test, [[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]]))
get_test = hk_data[:]
assert(array_util.array_comp(get_test, [[[10.0, 20.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0]]]))
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_frf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'flow15_flow.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')]
assert array_util.array_comp(budget_frf_valid, budget_frf)
# clean up
sim.delete_output_files()
return
def test003_gwfs_disv():
# init paths
test_ex_name = 'test003_gwfs_disv'
model_name = 'gwf_1'
pth = os.path.join('..', 'examples', 'data', 'mf6', test_ex_name)
run_folder = os.path.join(cpth, test_ex_name)
if not os.path.isdir(run_folder):
os.makedirs(run_folder)
save_folder = os.path.join(run_folder, 'temp')
if not os.path.isdir(save_folder):
os.makedirs(save_folder)
expected_output_folder = os.path.join(pth, 'expected_output')
expected_head_file_a = os.path.join(expected_output_folder, 'model_unch.hds')
expected_head_file_b = os.path.join(expected_output_folder, 'model_adj.hds')
expected_cbc_file_a = os.path.join(expected_output_folder, 'model_unch.cbc')
expected_cbc_file_b = os.path.join(expected_output_folder, 'model_adj.cbc')
array_util = PyListUtil()
# load simulation
sim = MFSimulation.load(model_name, 'mf6', exe_name, pth)
# make temp folder to save simulation
sim.simulation_data.mfpath.set_sim_path(run_folder)
# write simulation to new location
sim.simulation_data.max_columns_of_data = 10
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_a)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(budget_obj.get_data(text=' FLOW JA FACE', full3D=True))
head_file = os.path.join(os.getcwd(), expected_head_file_a)
head_new = os.path.join(run_folder, 'model.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')]
assert array_util.array_comp(budget_fjf_valid, budget_frf)
model = sim.get_model(model_name)
model.export('{}/{}.shp'.format(pth, test_ex_name))
# change some settings
chd_head_left = model.get_package('CHD_LEFT')
chd_left_period = chd_head_left.stress_period_data.array
chd_left_period[0][4][1] = 15.0
chd_head_right = model.get_package('CHD_RIGHT')
chd_right_period = chd_head_right.stress_period_data
chd_right_data = chd_right_period.get_data(0)
chd_right_data_slice = chd_right_data[3:10]
chd_right_period.set_data(chd_right_data_slice, 0)
# write simulation again
sim.simulation_data.mfpath.set_sim_path(save_folder)
sim.write_simulation()
if run:
# run simulation
sim.run_simulation()
# get expected results
budget_file = os.path.join(os.getcwd(), expected_cbc_file_b)
budget_obj = bf.CellBudgetFile(budget_file, precision='double')
budget_fjf_valid = np.array(budget_obj.get_data(text='FLOW JA FACE', full3D=True))
# compare output to expected results
head_file = os.path.join(os.getcwd(), expected_head_file_b)
head_new = os.path.join(save_folder, 'model.hds')
assert pymake.compare_heads(None, None, files1=head_file, files2=head_new)
budget_frf = sim.simulation_data.mfdata[(model_name, 'CBC', 'FLOW-JA-FACE')]
assert array_util.array_comp(budget_fjf_valid, budget_frf)
# clean up
sim.delete_output_files()
return