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