diff --git a/tests/test_015_gwf_output.py b/tests/test_015_gwf_output.py index 7fa41eaf..8083bee5 100644 --- a/tests/test_015_gwf_output.py +++ b/tests/test_015_gwf_output.py @@ -46,7 +46,7 @@ def test_create_small_model_grid_only(tmpdir, model_name="test"): nlmod.gwf.dis(ds, gwf) # create node property flow - nlmod.gwf.npf(ds, gwf, save_flows=True) + nlmod.gwf.npf(ds, gwf, save_flows=True, save_specific_discharge=True) # Create the initial conditions package nlmod.gwf.ic(ds, gwf, starting_head=1.0) @@ -76,6 +76,9 @@ def test_create_small_model_grid_only(tmpdir, model_name="test"): da = get_budget_da("CHD", ds=None, gwf=gwf, fname=None) # gwf fname_cbc = os.path.join(ds.model_ws, ds.model_name + ".cbc") get_budget_da("CHD", ds=None, gwf=None, fname=fname_cbc, grbfile=grbfile) # fname + get_budget_da( + "DATA-SPDIS", column="qz", ds=None, gwf=None, fname=fname_cbc, grbfile=grbfile + ) # fname # unstructured ds_unstr = refine( @@ -103,7 +106,7 @@ def test_create_small_model_grid_only(tmpdir, model_name="test"): nlmod.gwf.dis(ds_unstr, gwf_unstr) # create node property flow - nlmod.gwf.npf(ds_unstr, gwf_unstr) + nlmod.gwf.npf(ds_unstr, gwf_unstr, save_flows=True, save_specific_discharge=True) # Create the initial conditions package nlmod.gwf.ic(ds_unstr, gwf_unstr, starting_head=1.0) @@ -134,6 +137,9 @@ def test_create_small_model_grid_only(tmpdir, model_name="test"): da = get_budget_da( "CHD", ds=None, gwf=None, fname=fname_cbc, grbfile=grbfile ) # fname + _ = get_budget_da( + "DATA-SPDIS", column="qz", ds=None, gwf=None, fname=fname_cbc, grbfile=grbfile + ) # fname def test_get_heads_da_from_file_structured_no_grb():