Updated plot tests and added images

test/plot_tests/plot_all.ipynb

@@ -15,9 +15,9 @@
     "# Modify these to point to your own files\n",
     "WRF_DIRECTORY = \"/Users/ladwig/Documents/wrf_files/wrf_vortex_multi/moving_nest\"\n",
     "\n",
-    "WRF_FILES = [\"wrfout_d01_2005-08-28_00:00:00\",\n",
-    "             \"wrfout_d01_2005-08-28_12:00:00\",\n",
-    "             \"wrfout_d01_2005-08-29_00:00:00\"]\n",
+    "WRF_FILES = [\"wrfout_d02_2005-08-28_00:00:00\",\n",
+    "             \"wrfout_d02_2005-08-28_12:00:00\",\n",
+    "             \"wrfout_d02_2005-08-29_00:00:00\"]\n",
     "\n",
     "\n",
     "# Do not modify the code below this line\n",
@@ -67,6 +67,16 @@
     "    global _WRF_FILES\n",
     "    return _WRF_FILES\n",
     "\n",
+    "def save_fig(diagname):\n",
+    "    f = single_wrf_file()\n",
+    "    if f.find(\"_d01_\") > 0:\n",
+    "        fout = \"{}.png\".format(os.path.join(os.path.abspath(\".\"), \"d01\", diagname))\n",
+    "    else:\n",
+    "        fout = \"{}.png\".format(os.path.join(os.path.abspath(\".\"), \"d02\", diagname))\n",
+    "    \n",
+    "    matplotlib.pyplot.savefig(fout)\n",
+    "        \n",
+    "\n",
     "print(\"All tests passed!\")"
    ]
   },
@@ -243,6 +253,7 @@
     "    # Get the cartopy object and the lat,lon coords\n",
     "    cart_proj = get_cartopy(diag)\n",
     "    lats, lons = latlon_coords(diag)\n",
+    "    \n",
     "\n",
     "    # Create a figure and get the GetAxes object\n",
     "    fig = pyplot.figure(figsize=(10, 7.5))\n",
@@ -280,6 +291,10 @@
     "    cb = pyplot.colorbar(ax=ax, shrink=.99)\n",
     "    \n",
     "    pyplot.title(diagname)\n",
+    "    \n",
+    "    # Uncomment this to build the sample images\n",
+    "    #save_fig(diagname)\n",
+    "    \n",
     "    pyplot.show()\n",
     "\n",
     "\n"