Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

update return types for capacity_update_configuration #6489

Merged
3 commits merged into from
Feb 16, 2024
Merged

update return types for capacity_update_configuration #6489

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0c219f7
update return types for capacity_update_config
Feb 15, 2024
18331d4
format
Feb 15, 2024
4a315fe
Merge branch 'master' into md/update-capacity-parser-return-types
Feb 16, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 7 additions & 41 deletions scripts/update_capacity_configuration.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,12 +84,12 @@ def update_zone_capacity_config(zone_key: ZoneKey, data: dict) -> None:
capacity = _new_zone_config["capacity"]

if all(isinstance(capacity[m], float | int) for m in capacity.keys()):
# TODO: this is temporry as it handles the case of the deprecated system where capacity is a single value. It will be removed in the future
capacity = data
# TODO: this is temporary as it handles the case of the deprecated system where capacity is a single value. It will be removed in the future
capacity = {key: [value] for key, value in data.items()}
else:
capacity = generate_zone_capacity_config(capacity, data)
else:
capacity = data
capacity = {key: [value] for key, value in data.items()}

_new_zone_config["capacity"] = capacity

Expand All @@ -111,48 +111,18 @@ def generate_zone_capacity_config(
updated_capacity_config = deepcopy(capacity_config)
for mode in existing_capacity_modes:
if isinstance(capacity_config[mode], float | int):
updated_capacity_config[mode] = data[mode]
elif isinstance(capacity_config[mode], dict):
updated_capacity_config[mode] = generate_zone_capacity_dict(
mode, capacity_config, data
)
updated_capacity_config[mode] = [data[mode]]
elif isinstance(capacity_config[mode], list):
updated_capacity_config[mode] = generate_zone_capacity_list(
mode, capacity_config, data
)

new_modes = [m for m in data if m not in capacity_config]
for mode in new_modes:
updated_capacity_config[mode] = data[mode]
updated_capacity_config[mode] = [data[mode]]
return updated_capacity_config


def update_capacity_dict_if_value_already_exists(
mode: str, capacity_config: dict[str, Any], new_capacity: dict[str, Any]
) -> dict[str, Any]:
"""Updates the capacity config for a zone if the capacity config is a list and the value already exists.
This function ensures that we don't add the same value to the config over and over and that we can backfill and get the oldest value for which the capacity is valid.
"""
if new_capacity[mode]["datetime"] > capacity_config[mode]["datetime"]:
# if the associated datetime is more recent than the existing one, we don't add it
return capacity_config[mode]
# if the associated datetime is older than the existing one, we replace the datetime of the item
return new_capacity[mode]


def generate_zone_capacity_dict(
mode: str, capacity_config: dict[str, Any], new_capacity: dict[str, Any]
) -> dict[str, Any]:
"""Generate the updated capacity config for a zone if the capacity config is a dict"""
existing_capacity = capacity_config[mode]
if existing_capacity["value"] == new_capacity[mode]["value"]:
return update_capacity_dict_if_value_already_exists(
mode, capacity_config, new_capacity
)
else:
return [existing_capacity] + [new_capacity[mode]]


def update_capacity_list_if_value_already_exists(
mode: str, capacity_config: list[dict[str, Any]], new_capacity: dict[str, Any]
) -> list[dict[str, Any]]:
Expand Down Expand Up @@ -235,17 +205,13 @@ def generate_aggregated_capacity_config(
f"{parent_zone} capacity could not be updated because all capacity configs must have the same type"
)

if check_capacity_config_type(mode_capacity_configs, dict):
parent_capacity_config[mode] = generate_aggregated_capacity_config_dict(
mode_capacity_configs, parent_zone
)
elif check_capacity_config_type(mode_capacity_configs, list):
if check_capacity_config_type(mode_capacity_configs, list):
parent_capacity_config[mode] = generate_aggregated_capacity_config_list(
mode_capacity_configs, parent_zone
)
else:
raise ValueError(
f"{parent_zone} capacity could not be updated because all capacity configs must be a dict or a list"
f"{parent_zone} capacity could not be updated because all capacity configs must be a list"
)
return parent_capacity_config

Expand Down
59 changes: 21 additions & 38 deletions tests/test_update_capacity_configuration.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,19 +5,20 @@
generate_aggregated_capacity_config_list,
generate_zone_capacity_config,
generate_zone_capacity_list,
update_capacity_dict_if_value_already_exists,
)


class updateCapacityConfigurationTestCase(unittest.TestCase):
def test_capacity_config(self):
capacity_config = {
"wind": 1,
"solar": {
"datetime": "2022-01-01",
"source": "abc",
"value": 2,
},
"solar": [
{
"datetime": "2022-01-01",
"source": "abc",
"value": 2,
}
],
"biomass": [
{
"datetime": "2022-01-01",
Expand Down Expand Up @@ -72,11 +73,13 @@ def test_capacity_config(self):
}

expected = {
"wind": {
"datetime": "2023-01-01",
"source": "abc",
"value": 3,
},
"wind": [
{
"datetime": "2023-01-01",
"source": "abc",
"value": 3,
}
],
"solar": [
{
"datetime": "2022-01-01",
Expand All @@ -101,11 +104,13 @@ def test_capacity_config(self):
"value": 5,
},
],
"hydro": {
"datetime": "2023-01-01",
"source": "abc",
"value": 6,
},
"hydro": [
{
"datetime": "2023-01-01",
"source": "abc",
"value": 6,
}
],
"unknown": [
{
"datetime": "2022-01-01",
Expand All @@ -124,7 +129,6 @@ def test_capacity_config(self):
},
],
}

self.assertEqual(generate_zone_capacity_config(capacity_config, data), expected)

def test_generate_zone_capacity_list(self):
Expand Down Expand Up @@ -368,24 +372,3 @@ def test_generate_aggregated_capacity_config_list(self):
assert len(expected) == len(updated_capacity)
assert expected[0] in updated_capacity
assert expected[1] in updated_capacity

def test_update_capacity_dict_if_value_already_exists(self):
capacity_config = {
"biomass": {"datetime": "2022-01-01", "source": "abc", "value": 3}
}

data = {"biomass": {"datetime": "2023-01-01", "source": "abc", "value": 3}}

self.assertEqual(
update_capacity_dict_if_value_already_exists(
"biomass", capacity_config, data
),
capacity_config["biomass"],
)

self.assertEqual(
update_capacity_dict_if_value_already_exists(
"biomass", data, capacity_config
),
capacity_config["biomass"],
)