test_capacity.py

from datetime import datetime

import pytest

from electricitymap.contrib.config.capacity import (
    CapacityData,
    get_capacity_data,
    get_capacity_data_with_source,
)


def test_get_capacity_data():
    capacity_data = {}
    capacity_data_1 = {"solar": 3, "wind": 4, "coal": None}
    capacity_data_2 = {
        "coal": {"datetime": "2022-01-01", "value": 5, "source": "abc"},
        "gas": {"datetime": "2022-01-01", "value": 6, "source": "abc"},
        "solar": {"datetime": "2022-01-01", "value": None, "source": "abc"},
    }
    capacity_data_3 = {
        "coal": [
            {"datetime": "2022-01-01", "value": 5, "source": "abc"},
            {"datetime": "2023-06-01", "value": 8, "source": "abc"},
        ],
        "gas": [
            {"datetime": "2022-01-01", "value": 6, "source": "abc"},
            {"datetime": "2023-06-01", "value": 7, "source": "abc"},
        ],
    }
    assert get_capacity_data(capacity_data, datetime(2023, 1, 1)) == {}
    assert get_capacity_data(capacity_data_1, datetime(2023, 1, 1)) == {
        "solar": 3,
        "wind": 4,
        "coal": None,
    }
    assert get_capacity_data(capacity_data_2, datetime(2022, 6, 1)) == {
        "coal": 5,
        "gas": 6,
        "solar": None,
    }
    assert get_capacity_data(capacity_data_3, datetime(2023, 6, 1)) == {
        "coal": 8,
        "gas": 7,
    }

    with pytest.raises(ValueError):
        get_capacity_data(capacity_data_2, datetime(2023, 0, 1))


def test_get_capacity_from_list():
    mode_capacity = [
        {"datetime": "2022-01-01", "value": 3, "source": "abc"},
        {"datetime": "2023-01-01", "value": 4, "source": "abc"},
    ]

    capacity_dt = datetime(2022, 1, 1)
    for item in mode_capacity:
        if datetime.fromisoformat(item["datetime"]).date() == capacity_dt.date():
            capacity = item["value"]

    assert capacity == 3


def test_get_capacity_with_source():
    capacity_config = {
        "solar": {"datetime": "2023-10-01", "value": 3, "source": "abc"},
        "wind": [
            {"datetime": "2023-10-01", "value": 4, "source": "abc"},
            {"datetime": "2023-11-01", "value": 5, "source": "abc"},
        ],
    }

    capacity_data = get_capacity_data_with_source(
        capacity_config, datetime(2023, 10, 1)
    )

    assert capacity_data == {
        "solar": CapacityData(3, "abc"),
        "wind": CapacityData(4, "abc"),
    }

    assert get_capacity_data_with_source(capacity_config, datetime(2023, 11, 1)) == {
        "solar": CapacityData(3, "abc"),
        "wind": CapacityData(5, "abc"),
    }