add test for column from budget file

- for column qz from DATA_SPDIS
gwmod · Nov 2, 2023 · 6e3de76 · 6e3de76
1 parent 2fb79d3
commit 6e3de76
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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():