diff --git a/awsm/tests/basins/Gold-NetCDF-Compare.ipynb b/awsm/tests/basins/Gold-NetCDF-Compare.ipynb index e559953..f5624fd 100644 --- a/awsm/tests/basins/Gold-NetCDF-Compare.ipynb +++ b/awsm/tests/basins/Gold-NetCDF-Compare.ipynb @@ -45,7 +45,11 @@ "id": "5070313f-db14-438e-8c18-dd2ce0b95621", "metadata": {}, "source": [ - "### Test basin" + "### Test basin\n", + "Pick one of the available below. This will ensure proper comparison path setup:\n", + "* Lakes: Lakes with HRRR\n", + "* Lakes_lidar: Lakes with HRRR and lidar update\n", + "* RME: RME with station data" ] }, { @@ -55,7 +59,21 @@ "metadata": {}, "outputs": [], "source": [ - "basin_name = \"Lakes\"" + "basin_name = \"Lakes_lidar\"" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "277416bc-035f-42f6-941e-813843bf5b42", + "metadata": {}, + "outputs": [], + "source": [ + "basins = {\n", + " \"Lakes\": (\"Lakes\", \"gold_hrrr\"),\n", + " \"Lakes_lidar\": (\"Lakes\", \"gold_hrrr_update\"),\n", + " \"RME\": (\"RME\", \"gold\"),\n", + "}" ] }, { @@ -110,19 +128,30 @@ "metadata": {}, "outputs": [], "source": [ - "filename = \"model_lidar_change.nc\"\n", - "variables = lidar_change" + "filename = \"em.nc\"\n", + "variables = em_variables" + ] + }, + { + "cell_type": "markdown", + "id": "b5dcab09-9890-409f-8b92-443422919d1a", + "metadata": {}, + "source": [ + "## NOTE\n", + "When changing between Lakes and Lakes with lidar update, ensure that the gold path is updated below\n", + "* Lakes: `gold_hrrr`\n", + "* Lakes with lidar: `gold_hrrr_update`" ] }, { "cell_type": "code", "execution_count": null, - "id": "eac36c8e-4863-4733-9c3c-a7854eb70f02", + "id": "86a76ade-3f51-415f-829a-a7304f2af080", "metadata": {}, "outputs": [], "source": [ - "old_nc = xr.open_dataset(f\"{basin_name}/gold_hrrr_update/{filename}\").drop_vars(\"projection\")\n", - "new_nc_file = list(Path('.').glob(f\"{basin_name}/output/*/*/*/*/{filename}\"))[0]\n", + "old_nc = xr.open_dataset(f\"{basins[basin_name][0]}/{basins[basin_name][1]}/{filename}\").drop_vars(\"projection\")\n", + "new_nc_file = list(Path('.').glob(f\"{basins[basin_name][0]}/output/*/*/*/*/{filename}\"))[0]\n", "new_nc = xr.open_dataset(new_nc_file).drop_vars(\"projection\")" ] }, @@ -167,6 +196,8 @@ " \n", " plots.set_titles(\"{value}\")\n", " plots.fig.suptitle(f\"Variable: {variable}\")\n", + "\n", + " plots.fig.savefig(f\"{variable}_xy.png\", dpi=120)\n", " \n", " return plots" ] @@ -221,7 +252,9 @@ " \n", " for i in range(num_times):\n", " data[variable].isel(time=i).plot.hist(ax=axes[i], bins=30, color='skyblue')\n", - " axes[i].set_xlabel(\"\")" + " axes[i].set_xlabel(\"\")\n", + "\n", + " fig.savefig(f\"{variable}_hist.png\", dpi=120)" ] }, { @@ -238,7 +271,7 @@ { "cell_type": "code", "execution_count": null, - "id": "e5b4b7d9-6c11-4ed3-b015-0bd6fde9b5d1", + "id": "cbfaa1be-e013-4301-a711-b8cb43e93fdf", "metadata": {}, "outputs": [], "source": [] diff --git a/awsm/tests/basins/Lakes/gold_hrrr/em.nc b/awsm/tests/basins/Lakes/gold_hrrr/em.nc index 19849af..14443fc 100644 Binary files a/awsm/tests/basins/Lakes/gold_hrrr/em.nc and b/awsm/tests/basins/Lakes/gold_hrrr/em.nc differ diff --git a/awsm/tests/basins/Lakes/gold_hrrr_update/em.nc b/awsm/tests/basins/Lakes/gold_hrrr_update/em.nc index 6d10736..716d6c7 100644 Binary files a/awsm/tests/basins/Lakes/gold_hrrr_update/em.nc and b/awsm/tests/basins/Lakes/gold_hrrr_update/em.nc differ