From 6284cf7e8b210bf49c3fc6dd70b745b7739c68c4 Mon Sep 17 00:00:00 2001 From: Jonas Hoersch Date: Thu, 7 Feb 2019 17:41:01 +0100 Subject: [PATCH] add_electicity: Report countries missing from the hydro inflow time-series --- scripts/add_electricity.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/scripts/add_electricity.py b/scripts/add_electricity.py index f5be25c6..b46185d1 100644 --- a/scripts/add_electricity.py +++ b/scripts/add_electricity.py @@ -246,8 +246,12 @@ def attach_hydro(n, costs, ppl): dist_key = ppl.loc[ppl.has_inflow, 'p_nom'].groupby(country).transform(normed) with xr.open_dataarray(snakemake.input.profile_hydro) as inflow: + inflow_countries = pd.Index(country.loc[ppl.has_inflow].values) + assert len(inflow_countries.unique().difference(inflow.indexes['countries'])) == 0, \ + "'{}' is missing inflow time-series for at least one country: {}".format(snakemake.input.profile_hydro, ", ".join(inflow_countries.unique().difference(inflow.indexes['countries']))) + inflow_t = ( - inflow.sel(countries=country.loc[ppl.has_inflow].values) + inflow.sel(countries=inflow_countries) .rename({'countries': 'name'}) .assign_coords(name=ppl.index[ppl.has_inflow]) .transpose('time', 'name')