def test_setup_from_yaml_issue(project_root_path, remake_top):
cfg = MF6model.load_cfg(
Path(project_root_path) / 'examples/pleasant_lgr_parent.yml')
lgr_test_path = Path(project_root_path) / 'examples/pleasant_lgr'
shutil.rmtree(lgr_test_path, ignore_errors=True)
keep_keys = {
'simulation', 'model', 'parent', 'setup_grid', 'dis', 'tdis',
'intermediate_data', 'postprocessing', 'filename'
}
new_cfg = {k: v for k, v in cfg.items() if k in keep_keys}
new_cfg['model']['packages'] = ['dis']
new_cfg['dis']['remake_top'] = remake_top
del new_cfg['setup_grid']['lgr']
MF6model.setup_from_cfg(new_cfg)
def pleasant_lgr_stand_alone_parent(pleasant_lgr_test_cfg_path, tmpdir):
"""Stand-alone version of lgr parent model for comparing with LGR results.
"""
# Edit the configuration file before the file paths within it are converted to absolute
# (model.load_cfg converts the file paths)
cfg = load(pleasant_lgr_test_cfg_path)
del cfg['setup_grid']['lgr']
cfg['simulation']['sim_ws'] = os.path.join(tmpdir,
'pleasant_lgr_just_parent')
# save out the edited configuration file
path, fname = os.path.split(pleasant_lgr_test_cfg_path)
new_file = os.path.join(path, 'pleasant_lgr_just_parent.yml')
dump(new_file, cfg)
# load in the edited configuration file, converting the paths to absolute
cfg = MF6model.load_cfg(new_file)
# add some stuff just for the tests
cfg['gisdir'] = os.path.join(cfg['simulation']['sim_ws'], 'gis')
m = MF6model.setup_from_cfg(cfg)
m.write_input()
#if hasattr(m, 'sfr'):
# sfr_package_filename = os.path.join(m.model_ws, m.sfr.filename)
# m.sfrdata.write_package(sfr_package_filename,
# version='mf6'
# )
return m
def pleasant_lgr_setup_from_yaml(pleasant_lgr_cfg):
m = MF6model.setup_from_cfg(pleasant_lgr_cfg)
m.write_input()
for model in m, m.inset['plsnt_lgr_inset']:
if hasattr(model, 'sfr'):
sfr_package_filename = os.path.join(model.model_ws, model.sfr.filename)
model.sfrdata.write_package(sfr_package_filename,
version='mf6'
)
return m
def pleasant_lgr_setup_from_yaml(pleasant_lgr_cfg):
m = MF6model.setup_from_cfg(pleasant_lgr_cfg)
m.write_input()
#for model in m, m.inset['plsnt_lgr_inset']:
# if hasattr(model, 'sfr'):
# sfr_package_filename = os.path.join(model.model_ws, model.sfr.filename)
# model.sfrdata.write_package(sfr_package_filename,
# version='mf6',
# options=['save_flows',
# 'BUDGET FILEOUT {}.sfr.cbc'.format(model.name),
# 'STAGE FILEOUT {}.sfr.stage.bin'.format(model.name),
# 'mover'
# ]
# )
return m
def pleasant_lgr_setup_from_yaml(pleasant_lgr_cfg):
m = MF6model.setup_from_cfg(pleasant_lgr_cfg)
m.write_input()
return m
