scenario_builder.storages.scenario_storages()

Complexity

Conditions

1

Size

Total Lines	28
Code Lines	3

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

"""Create a basic scenario from the internal data structure.
def scenario_storages(regions, year, name):
    """
    Fetch storage, pump and turbine capacity and their efficiency of
    hydroelectric storages for each deflex region.

    Parameters
    ----------
    regions
    year
    name

    Returns
    -------

    Examples
    --------
    >>> from reegis import geometries
    >>> fs=geometries.get_federal_states_polygon()
    >>> deflex_storages=scenario_storages(fs, 2012, "de17")
    >>> list(deflex_storages.index.get_level_values(0))
    ['BW', 'BY', 'HE', 'NI', 'NW', 'SH', 'SN', 'ST', 'TH']
    >>> int(deflex_storages.loc[("TH", "phes"), "turbine"])
    1522
    >>> int(deflex_storages.loc[("TH", "phes"), "energy"])
    12115
    """
    stor = storages.pumped_hydroelectric_storage_by_region(regions, year, name)
    return pd.concat([stor], keys=["phes"]).swaplevel(0, 1)