mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2026-07-19 08:18:50 +00:00
eb9e966de9
FAstAPI is an async framework. Data may be imported and exported, load and save, set and get asynchronously. Prevent interleaving data operations to corrupt the data. In the previous design sync and async data access was intermixed leading to data corruption. The basic data classes DataSequence and DataContainer and the derived classes like Provider and Measurement now are async. Data access is protected by several async locks. To support the async design of the data classes the database interface became async. The energy management is also adapted to the new async design. Optimization is still off-loaded to another thread, but the prepration for the optimization and the post optimization actions now follow the async design. Adapter operations are now also protected by async locks. Tests were adapted to the async design and new tests were created. Besides this major fix several other improvements and fixes are included in this PR. * fix: key_to_dict/list/array only regard data records with key value set. Before the exclusion of no value data records was only done if the dropna flag was set. * fix: test for visual result pdf generation Due to updates in the library the generated charts text was a little bit different. Adapt the test to create the comaprison pdf in the test data durectory and update the reference pdf. * chore: Remove MutableMapping from DataSequence and DataContainer. Mutable Mapping does not fit to the now async design. * chore: Add NoDB database backend This backend implements the full database backend interface but performs no actual persistence. It is intended for configurations where database persistence is disabled (`provider=None`). * chore: Improve measurement data import testing with real world scenarios. Added two new endpoints to support testing. * chore: Add mermaid to supported documentation tools * chore: Add documentation about async design * chore: Add documentation about generic data handling Covers the basics of measurement and prediction time series data handling. * chore: Add empty lines around markdown lists. * chore: sync pre-commit config to updated package versions Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>
118 lines
4.4 KiB
Python
118 lines
4.4 KiB
Python
import asyncio
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import numpy.testing as npt
|
|
import pytest
|
|
|
|
from akkudoktoreos.core.coreabc import get_ems
|
|
from akkudoktoreos.prediction.elecpriceimport import ElecPriceImport
|
|
from akkudoktoreos.utils.datetimeutil import compare_datetimes, to_datetime, to_duration
|
|
|
|
DIR_TESTDATA = Path(__file__).absolute().parent.joinpath("testdata")
|
|
|
|
FILE_TESTDATA_ELECPRICEIMPORT_1_JSON = DIR_TESTDATA.joinpath("import_input_1.json")
|
|
|
|
|
|
@pytest.fixture
|
|
def provider(sample_import_1_json, config_eos):
|
|
"""Fixture to create a ElecPriceProvider instance."""
|
|
settings = {
|
|
"elecprice": {
|
|
"provider": "ElecPriceImport",
|
|
"elecpriceimport": {
|
|
"import_file_path": str(FILE_TESTDATA_ELECPRICEIMPORT_1_JSON),
|
|
"import_json": json.dumps(sample_import_1_json),
|
|
},
|
|
}
|
|
}
|
|
config_eos.merge_settings_from_dict(settings)
|
|
provider = ElecPriceImport()
|
|
assert provider.enabled()
|
|
return provider
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_import_1_json():
|
|
"""Fixture that returns sample forecast data report."""
|
|
with FILE_TESTDATA_ELECPRICEIMPORT_1_JSON.open("r", encoding="utf-8", newline=None) as f_res:
|
|
input_data = json.load(f_res)
|
|
return input_data
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestElecPriceImport:
|
|
# ------------------------------------------------
|
|
# General forecast
|
|
# ------------------------------------------------
|
|
|
|
|
|
def test_singleton_instance(self, provider):
|
|
"""Test that ElecPriceForecast behaves as a singleton."""
|
|
another_instance = ElecPriceImport()
|
|
assert provider is another_instance
|
|
|
|
|
|
def test_invalid_provider(self, provider, config_eos):
|
|
"""Test requesting an unsupported provider."""
|
|
settings = {
|
|
"elecprice": {
|
|
"provider": "<invalid>",
|
|
"elecpriceimport": {
|
|
"import_file_path": str(FILE_TESTDATA_ELECPRICEIMPORT_1_JSON),
|
|
},
|
|
}
|
|
}
|
|
with pytest.raises(ValueError, match="not a valid electricity price provider"):
|
|
config_eos.merge_settings_from_dict(settings)
|
|
|
|
|
|
# ------------------------------------------------
|
|
# Import
|
|
# ------------------------------------------------
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"start_datetime, from_file",
|
|
[
|
|
("2024-11-10 00:00:00", True), # No DST in Germany
|
|
("2024-08-10 00:00:00", True), # DST in Germany
|
|
("2024-03-31 00:00:00", True), # DST change in Germany (23 hours/ day)
|
|
("2024-10-27 00:00:00", True), # DST change in Germany (25 hours/ day)
|
|
("2024-11-10 00:00:00", False), # No DST in Germany
|
|
("2024-08-10 00:00:00", False), # DST in Germany
|
|
("2024-03-31 00:00:00", False), # DST change in Germany (23 hours/ day)
|
|
("2024-10-27 00:00:00", False), # DST change in Germany (25 hours/ day)
|
|
],
|
|
)
|
|
async def test_import(self, provider, sample_import_1_json, start_datetime, from_file, config_eos):
|
|
"""Test fetching forecast from Import."""
|
|
key = "elecprice_marketprice_wh"
|
|
ems_eos = get_ems()
|
|
ems_eos.set_start_datetime(to_datetime(start_datetime, in_timezone="Europe/Berlin"))
|
|
if from_file:
|
|
config_eos.elecprice.elecpriceimport.import_json = None
|
|
assert config_eos.elecprice.elecpriceimport.import_json is None
|
|
else:
|
|
config_eos.elecprice.elecpriceimport.import_file_path = None
|
|
assert config_eos.elecprice.elecpriceimport.import_file_path is None
|
|
await provider.delete_by_datetime(start_datetime=None, end_datetime=None)
|
|
|
|
# Call the method
|
|
await provider.update_data()
|
|
|
|
# Assert: Verify the result is as expected
|
|
assert provider.ems_start_datetime is not None
|
|
assert provider.total_hours is not None
|
|
assert compare_datetimes(provider.ems_start_datetime, ems_eos.start_datetime).equal
|
|
|
|
expected_values = sample_import_1_json[key]
|
|
result_values = await provider.key_to_array(
|
|
key=key,
|
|
start_datetime=provider.ems_start_datetime,
|
|
end_datetime=provider.ems_start_datetime + to_duration(f"{len(expected_values)} hours"),
|
|
interval=to_duration("1 hour"),
|
|
)
|
|
# Allow for some difference due to value calculation on DST change
|
|
npt.assert_allclose(result_values, expected_values, rtol=0.001)
|