mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2026-08-31 12:46:38 +00:00
Add SMARD quarter-hour price provider
This commit is contained in:
@@ -0,0 +1,77 @@
|
||||
# ruff: noqa: S101
|
||||
|
||||
import json
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from akkudoktoreos.core.coreabc import get_ems
|
||||
from akkudoktoreos.prediction.elecprice import ElecPriceCommonSettings
|
||||
from akkudoktoreos.prediction.elecpricesmard import ElecPriceSMARD
|
||||
from akkudoktoreos.utils.datetimeutil import to_datetime
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def provider(config_eos):
|
||||
"""Configure and return the direct SMARD singleton provider."""
|
||||
config_eos.elecprice = ElecPriceCommonSettings(provider="ElecPriceSMARD")
|
||||
provider = ElecPriceSMARD()
|
||||
provider.highest_orig_datetime = None
|
||||
get_ems().set_start_datetime(
|
||||
to_datetime("2026-07-27 00:00:00", in_timezone="Europe/Berlin")
|
||||
)
|
||||
return provider
|
||||
|
||||
|
||||
def _response(payload):
|
||||
response = Mock()
|
||||
response.content = json.dumps(payload).encode()
|
||||
response.raise_for_status.return_value = None
|
||||
return response
|
||||
|
||||
|
||||
@patch("akkudoktoreos.prediction.elecpricesmard.requests.get")
|
||||
def test_request_forecast_fetches_index_and_overlapping_chunks(mock_get, provider):
|
||||
"""SMARD index and weekly chunks are combined, sorted, and stripped of null values."""
|
||||
chunk_start = 1785103200000
|
||||
mock_get.side_effect = [
|
||||
_response({"timestamps": [chunk_start]}),
|
||||
_response(
|
||||
{
|
||||
"meta_data": {"version": 1, "created": 1785500527370},
|
||||
"series": [
|
||||
[1785103200000, 86.04],
|
||||
[1785106800000, None],
|
||||
[1785110400000, -1.25],
|
||||
],
|
||||
}
|
||||
),
|
||||
]
|
||||
|
||||
result = provider._request_forecast(
|
||||
start_date="2026-07-27", force_update=True
|
||||
)
|
||||
|
||||
assert result.unix_seconds == [1785103200, 1785110400]
|
||||
assert result.price == [86.04, -1.25]
|
||||
assert result.license_info == "CC BY 4.0 Bundesnetzagentur | SMARD.de"
|
||||
assert mock_get.call_count == 2
|
||||
assert mock_get.call_args_list[0].args[0].endswith("/4169/DE/index_quarterhour.json")
|
||||
assert mock_get.call_args_list[1].args[0].endswith(
|
||||
"/4169/DE/4169_DE_quarterhour_1785103200000.json"
|
||||
)
|
||||
|
||||
|
||||
def test_chunk_selection_includes_preceding_overlapping_chunk(provider):
|
||||
"""A range beginning mid-week includes the chunk that started before it."""
|
||||
index = provider._validate_index(
|
||||
json.dumps({"timestamps": [1000, 2000, 3000]}).encode()
|
||||
)
|
||||
start = to_datetime(2.5, in_timezone="UTC")
|
||||
end = to_datetime(3.5, in_timezone="UTC")
|
||||
|
||||
assert provider._chunk_timestamps(index, start, end) == [2000, 3000]
|
||||
|
||||
|
||||
def test_smard_provider_is_enabled(provider):
|
||||
assert provider.enabled()
|
||||
Reference in New Issue
Block a user