test_connect_invest.test_connect_invest()

Complexity

Conditions

2

Size

Total Lines	124
Code Lines	83

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

# -*- coding: utf-8 -*-
def test_connect_invest():
    date_time_index = pd.date_range("1/1/2012", periods=24 * 7, freq="h")

    es = EnergySystem(timeindex=date_time_index, infer_last_interval=True)

    # Read data file
    full_filename = os.path.join(
        os.path.dirname(__file__), "connect_invest.csv"
    )
    data = pd.read_csv(full_filename, sep=",")

    logging.info("Create oemof objects")

    # create electricity bus
    bel1 = Bus(label="electricity1")
    bel2 = Bus(label="electricity2")
    es.add(bel1, bel2)

    # create excess component for the electricity bus to allow overproduction
    es.add(components.Sink(label="excess_bel", inputs={bel2: Flow()}))
    es.add(
        components.Source(
            label="shortage", outputs={bel2: Flow(variable_costs=50000)}
        )
    )

    # create fixed source object representing wind power plants
    es.add(
        components.Source(
            label="wind",
            outputs={bel1: Flow(fix=data["wind"], nominal_capacity=1000000)},
        )
    )

    # create simple sink object representing the electrical demand
    es.add(
        components.Sink(
            label="demand",
            inputs={bel1: Flow(fix=data["demand_el"], nominal_capacity=1)},
        )
    )

    storage = components.GenericStorage(
        label="storage",
        inputs={bel1: Flow(variable_costs=10e10)},
        outputs={bel1: Flow(variable_costs=10e10)},
        loss_rate=0.00,
        initial_storage_level=0,
        invest_relation_input_capacity=1 / 6,
        invest_relation_output_capacity=1 / 6,
        inflow_conversion_factor=1,
        outflow_conversion_factor=0.8,
        nominal_capacity=Investment(ep_costs=0.2),
    )
    es.add(storage)

    line12 = components.Converter(
        label="line12",
        inputs={bel1: Flow()},
        outputs={bel2: Flow(nominal_capacity=Investment(ep_costs=20))},
    )
    es.add(line12)

    line21 = components.Converter(
        label="line21",
        inputs={bel2: Flow()},
        outputs={bel1: Flow(nominal_capacity=Investment(ep_costs=20))},
    )
    es.add(line21)

    om = Model(es)

    constraints.equate_variables(
        om,
        om.InvestmentFlowBlock.invest[line12, bel2, 0],
        om.InvestmentFlowBlock.invest[line21, bel1, 0],
        2,
    )
    constraints.equate_variables(
        om,
        om.InvestmentFlowBlock.invest[line12, bel2, 0],
        om.GenericInvestmentStorageBlock.invest[storage, 0],
    )

    # if tee_switch is true solver messages will be displayed
    logging.info("Solve the optimization problem")
    om.solve(solver="cbc", tee=True)

    # check if the new result object is working for custom components
    results = processing.results(om)

    my_results = dict()
    my_results["line12"] = (
        views.node(results, "line12")["scalars"]
        .loc[[(("line12", "electricity2"), "invest")]]
        .iloc[0]
    )

    my_results["line21"] = (
        views.node(results, "line21")["scalars"]
        .loc[[(("line21", "electricity1"), "invest")]]
        .iloc[0]
    )

    stor_res = views.node(results, "storage")["scalars"]
    my_results["storage_in"] = stor_res[
        [(("electricity1", "storage"), "invest")]
    ].iloc[0]
    my_results["storage"] = stor_res[[(("storage", "None"), "invest")]].iloc[0]
    my_results["storage_out"] = stor_res[
        [(("storage", "electricity1"), "invest")]
    ].iloc[0]

    connect_invest_dict = {
        "line12": 814705,
        "line21": 1629410,
        "storage": 814705,
        "storage_in": 135784,
        "storage_out": 135784,
    }

    for key in connect_invest_dict.keys():
        assert my_results[key] == pytest.approx(
            connect_invest_dict[key], abs=0.5
        )