From 6e3de76e62d812a8810a13041426f730b04e832c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dav=C3=ADd=20Brakenhoff?= Date: Thu, 2 Nov 2023 17:59:43 +0100 Subject: [PATCH] add test for column from budget file - for column qz from DATA_SPDIS --- tests/test_015_gwf_output.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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():