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 <[email protected]>
NOAA-OWP · Dec 3, 2021 · c74cf3f · c74cf3f
1 parent 2f13a68
commit c74cf3f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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")