mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
75987db9e1
* Add EOS_CONFIG_DIR to set config dir (relative path to EOS_DIR or
absolute path).
- config_folder_path read-only
- config_file_path read-only
* Default values to support app start with empty config:
- latitude/longitude (Berlin)
- optimization_ev_available_charge_rates_percent (null, so model
default value is used)
- Enable Akkudoktor electricity price forecast (docker-compose).
* Fix some endpoints (empty data, remove unused params, fix types).
* cacheutil: Use cache dir. Closes #240
* Support EOS_LOGGING_LEVEL environment variable to set log level.
* tests: All tests use separate temporary config
- Add pytest switch --check-config-side-effect to check user
config file existence after each test. Will also fail if user config
existed before test execution (but will only check after the test has
run).
Enable flag in github workflow.
- Globally mock platformdirs in config module. Now no longer required
to patch individually.
Function calls to config instance (e.g. merge_settings_from_dict)
were unaffected previously.
* Set Berlin as default location (default config/docker-compose).
110 lines
4.1 KiB
Python
110 lines
4.1 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from akkudoktoreos.core.ems import get_ems
|
|
from akkudoktoreos.prediction.elecpriceimport import ElecPriceImport
|
|
from akkudoktoreos.utils.datetimeutil import compare_datetimes, to_datetime
|
|
|
|
DIR_TESTDATA = Path(__file__).absolute().parent.joinpath("testdata")
|
|
|
|
FILE_TESTDATA_ELECPRICEIMPORT_1_JSON = DIR_TESTDATA.joinpath("import_input_1.json")
|
|
|
|
|
|
@pytest.fixture
|
|
def elecprice_provider(sample_import_1_json, config_eos):
|
|
"""Fixture to create a ElecPriceProvider instance."""
|
|
settings = {
|
|
"elecprice_provider": "ElecPriceImport",
|
|
"elecpriceimport_file_path": str(FILE_TESTDATA_ELECPRICEIMPORT_1_JSON),
|
|
"elecpriceimport_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 open(FILE_TESTDATA_ELECPRICEIMPORT_1_JSON, "r") as f_res:
|
|
input_data = json.load(f_res)
|
|
return input_data
|
|
|
|
|
|
# ------------------------------------------------
|
|
# General forecast
|
|
# ------------------------------------------------
|
|
|
|
|
|
def test_singleton_instance(elecprice_provider):
|
|
"""Test that ElecPriceForecast behaves as a singleton."""
|
|
another_instance = ElecPriceImport()
|
|
assert elecprice_provider is another_instance
|
|
|
|
|
|
def test_invalid_provider(elecprice_provider, config_eos):
|
|
"""Test requesting an unsupported elecprice_provider."""
|
|
settings = {
|
|
"elecprice_provider": "<invalid>",
|
|
"elecpriceimport_file_path": str(FILE_TESTDATA_ELECPRICEIMPORT_1_JSON),
|
|
}
|
|
config_eos.merge_settings_from_dict(settings)
|
|
assert not elecprice_provider.enabled()
|
|
|
|
|
|
# ------------------------------------------------
|
|
# 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)
|
|
],
|
|
)
|
|
def test_import(elecprice_provider, sample_import_1_json, start_datetime, from_file, config_eos):
|
|
"""Test fetching forecast from Import."""
|
|
ems_eos = get_ems()
|
|
ems_eos.set_start_datetime(to_datetime(start_datetime, in_timezone="Europe/Berlin"))
|
|
if from_file:
|
|
config_eos.elecpriceimport_json = None
|
|
assert config_eos.elecpriceimport_json is None
|
|
else:
|
|
config_eos.elecpriceimport_file_path = None
|
|
assert config_eos.elecpriceimport_file_path is None
|
|
elecprice_provider.clear()
|
|
|
|
# Call the method
|
|
elecprice_provider.update_data()
|
|
|
|
# Assert: Verify the result is as expected
|
|
assert elecprice_provider.start_datetime is not None
|
|
assert elecprice_provider.total_hours is not None
|
|
assert compare_datetimes(elecprice_provider.start_datetime, ems_eos.start_datetime).equal
|
|
values = sample_import_1_json["elecprice_marketprice"]
|
|
value_datetime_mapping = elecprice_provider.import_datetimes(
|
|
ems_eos.start_datetime, len(values)
|
|
)
|
|
for i, mapping in enumerate(value_datetime_mapping):
|
|
assert i < len(elecprice_provider.records)
|
|
expected_datetime, expected_value_index = mapping
|
|
expected_value = values[expected_value_index]
|
|
result_datetime = elecprice_provider.records[i].date_time
|
|
result_value = elecprice_provider.records[i]["elecprice_marketprice"]
|
|
|
|
# print(f"{i}: Expected: {expected_datetime}:{expected_value}")
|
|
# print(f"{i}: Result: {result_datetime}:{result_value}")
|
|
assert compare_datetimes(result_datetime, expected_datetime).equal
|
|
assert result_value == expected_value
|