From c74cf3fa5d534363a1a53d7637698bd6912eb2c4 Mon Sep 17 00:00:00 2001 From: Adam Wlostowski <66840445+awlostowski-noaa@users.noreply.github.com> Date: Fri, 3 Dec 2021 09:30:04 -0500 Subject: [PATCH] flowveldepth to pickle for testing (#503) * check statement * remove * save test * allow testing result to be printed from output.py * clear whitespace commit Co-authored-by: awlostowski-noaa --- src/nwm_routing/src/nwm_routing/output.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/nwm_routing/src/nwm_routing/output.py b/src/nwm_routing/src/nwm_routing/output.py index 1eba3e4aa..a7f2271cc 100644 --- a/src/nwm_routing/src/nwm_routing/output.py +++ b/src/nwm_routing/src/nwm_routing/output.py @@ -72,6 +72,7 @@ def nwm_output_generator( csv_output_folder = None rsrto = output_parameters.get("hydro_rst_output", None) chrto = output_parameters.get("chrtout_output", None) + test = output_parameters.get("test_output", None) if csv_output: csv_output_folder = output_parameters["csv_output"].get( @@ -80,7 +81,7 @@ def nwm_output_generator( csv_output_segments = csv_output.get("csv_output_segments", None) - if csv_output_folder or rsrto or chrto: + if csv_output_folder or rsrto or chrto or test: start = time.time() qvd_columns = pd.MultiIndex.from_product( @@ -110,6 +111,9 @@ def nwm_output_generator( LOG.debug("Constructing the FVD DataFrame took %s seconds." % (time.time() - start)) + if test: + flowveldepth.to_pickle(Path(test)) + if rsrto: LOG.info("- writing restart files")