mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
830af85fca
measurement:
- Add new measurement class to hold real world measurements.
- Handles load meter readings, grid import and export meter readings.
- Aggregates load meter readings aka. measurements to total load.
- Can import measurements from files, pandas datetime series,
pandas datetime dataframes, simple daetime arrays and
programmatically.
- Maybe expanded to other measurement values.
- Should be used for load prediction adaptions by real world
measurements.
core/coreabc:
- Add mixin class to access measurements
core/pydantic:
- Add pydantic models for pandas datetime series and dataframes.
- Add pydantic models for simple datetime array
core/dataabc:
- Provide DataImport mixin class for generic import handling.
Imports from JSON string and files. Imports from pandas datetime dataframes
and simple datetime arrays. Signature of import method changed to
allow import datetimes to be given programmatically and by data content.
- Use pydantic models for datetime series, dataframes, arrays
- Validate generic imports by pydantic models
- Provide new attributes min_datetime and max_datetime for DataSequence.
- Add parameter dropna to drop NAN/ None values when creating lists, pandas series
or numpy array from DataSequence.
config/config:
- Add common settings for the measurement module.
predictions/elecpriceakkudoktor:
- Use mean values of last 7 days to fill prediction values not provided by
akkudoktor.net (only provides 24 values).
prediction/loadabc:
- Extend the generic prediction keys by 'load_total_adjusted' for load predictions
that adjust the predicted total load by measured load values.
prediction/loadakkudoktor:
- Extend the Akkudoktor load prediction by load adjustment using measured load
values.
prediction/load_aggregator:
- Module removed. Load aggregation is now handled by the measurement module.
prediction/load_corrector:
- Module removed. Load correction (aka. adjustment of load prediction by
measured load energy) is handled by the LoadAkkudoktor prediction and
the generic 'load_mean_adjusted' prediction key.
prediction/load_forecast:
- Module removed. Functionality now completely handled by the LoadAkkudoktor
prediction.
utils/cacheutil:
- Use pydantic.
- Fix potential bug in ttl (time to live) duration handling.
utils/datetimeutil:
- Added missing handling of pendulum.DateTime and pendulum.Duration instances
as input. Handled before as datetime.datetime and datetime.timedelta.
utils/visualize:
- Move main to generate_example_report() for better testing support.
server/server:
- Added new configuration option server_fastapi_startup_server_fasthtml
to make startup of FastHTML server by FastAPI server conditional.
server/fastapi_server:
- Add APIs for measurements
- Improve APIs to provide or take pandas datetime series and
datetime dataframes controlled by pydantic model.
- Improve APIs to provide or take simple datetime data arrays
controlled by pydantic model.
- Move fastAPI server API to v1 for new APIs.
- Update pre v1 endpoints to use new prediction and measurement capabilities.
- Only start FastHTML server if 'server_fastapi_startup_server_fasthtml'
config option is set.
tests:
- Adapt import tests to changed import method signature
- Adapt server test to use the v1 API
- Extend the dataabc test to test for array generation from data
with several data interval scenarios.
- Extend the datetimeutil test to also test for correct handling
of to_datetime() providing now().
- Adapt LoadAkkudoktor test for new adjustment calculation.
- Adapt visualization test to use example report function instead of visualize.py
run as process.
- Removed test_load_aggregator. Functionality is now tested in test_measurement.
- Added tests for measurement module
docs:
- Remove sphinxcontrib-openapi as it prevents build of documentation.
"site-packages/sphinxcontrib/openapi/openapi31.py", line 305, in _get_type_from_schema
for t in schema["anyOf"]: KeyError: 'anyOf'"
Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>
213 lines
6.2 KiB
Python
213 lines
6.2 KiB
Python
from unittest.mock import patch
|
|
|
|
import numpy as np
|
|
import pendulum
|
|
import pytest
|
|
|
|
from akkudoktoreos.config.config import get_config
|
|
from akkudoktoreos.core.ems import get_ems
|
|
from akkudoktoreos.measurement.measurement import MeasurementDataRecord, get_measurement
|
|
from akkudoktoreos.prediction.loadakkudoktor import (
|
|
LoadAkkudoktor,
|
|
LoadAkkudoktorCommonSettings,
|
|
)
|
|
from akkudoktoreos.utils.datetimeutil import compare_datetimes, to_datetime, to_duration
|
|
|
|
config_eos = get_config()
|
|
ems_eos = get_ems()
|
|
|
|
|
|
@pytest.fixture
|
|
def load_provider():
|
|
"""Fixture to initialise the LoadAkkudoktor instance."""
|
|
settings = {
|
|
"load_provider": "LoadAkkudoktor",
|
|
"load_name": "Akkudoktor Profile",
|
|
"loadakkudoktor_year_energy": "1000",
|
|
}
|
|
config_eos.merge_settings_from_dict(settings)
|
|
return LoadAkkudoktor()
|
|
|
|
|
|
@pytest.fixture
|
|
def measurement_eos():
|
|
"""Fixture to initialise the Measurement instance."""
|
|
measurement = get_measurement()
|
|
load0_mr = 500
|
|
load1_mr = 500
|
|
dt = to_datetime("2024-01-01T00:00:00")
|
|
interval = to_duration("1 hour")
|
|
for i in range(25):
|
|
measurement.records.append(
|
|
MeasurementDataRecord(
|
|
date_time=dt,
|
|
measurement_load0_mr=load0_mr,
|
|
measurement_load1_mr=load1_mr,
|
|
)
|
|
)
|
|
dt += interval
|
|
load0_mr += 50
|
|
load1_mr += 50
|
|
assert compare_datetimes(measurement.min_datetime, to_datetime("2024-01-01T00:00:00")).equal
|
|
assert compare_datetimes(measurement.max_datetime, to_datetime("2024-01-02T00:00:00")).equal
|
|
return measurement
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_load_profiles_file(tmp_path):
|
|
"""Fixture to create a mock load profiles file."""
|
|
load_profiles_path = tmp_path / "load_profiles.npz"
|
|
np.savez(
|
|
load_profiles_path,
|
|
yearly_profiles=np.random.rand(365, 24), # Random load profiles
|
|
yearly_profiles_std=np.random.rand(365, 24), # Random standard deviation
|
|
)
|
|
return load_profiles_path
|
|
|
|
|
|
def test_loadakkudoktor_settings_validator():
|
|
"""Test the field validator for `loadakkudoktor_year_energy`."""
|
|
settings = LoadAkkudoktorCommonSettings(loadakkudoktor_year_energy=1234)
|
|
assert isinstance(settings.loadakkudoktor_year_energy, float)
|
|
assert settings.loadakkudoktor_year_energy == 1234.0
|
|
|
|
settings = LoadAkkudoktorCommonSettings(loadakkudoktor_year_energy=1234.56)
|
|
assert isinstance(settings.loadakkudoktor_year_energy, float)
|
|
assert settings.loadakkudoktor_year_energy == 1234.56
|
|
|
|
|
|
def test_loadakkudoktor_provider_id(load_provider):
|
|
"""Test the `provider_id` class method."""
|
|
assert load_provider.provider_id() == "LoadAkkudoktor"
|
|
|
|
|
|
@patch("akkudoktoreos.prediction.loadakkudoktor.Path")
|
|
@patch("akkudoktoreos.prediction.loadakkudoktor.np.load")
|
|
def test_load_data_from_mock(mock_np_load, mock_path, mock_load_profiles_file, load_provider):
|
|
"""Test the `load_data` method."""
|
|
# Mock path behavior to return the test file
|
|
mock_path.return_value.parent.parent.joinpath.return_value = mock_load_profiles_file
|
|
|
|
# Mock numpy load to return data similar to what would be in the file
|
|
mock_np_load.return_value = {
|
|
"yearly_profiles": np.ones((365, 24)),
|
|
"yearly_profiles_std": np.zeros((365, 24)),
|
|
}
|
|
|
|
# Test data loading
|
|
data_year_energy = load_provider.load_data()
|
|
assert data_year_energy is not None
|
|
assert data_year_energy.shape == (365, 2, 24)
|
|
|
|
|
|
def test_load_data_from_file(load_provider):
|
|
"""Test `load_data` loads data from the profiles file."""
|
|
data_year_energy = load_provider.load_data()
|
|
assert data_year_energy is not None
|
|
|
|
|
|
@patch("akkudoktoreos.prediction.loadakkudoktor.LoadAkkudoktor.load_data")
|
|
def test_update_data(mock_load_data, load_provider):
|
|
"""Test the `_update` method."""
|
|
mock_load_data.return_value = np.random.rand(365, 2, 24)
|
|
|
|
# Mock methods for updating values
|
|
ems_eos.set_start_datetime(pendulum.datetime(2024, 1, 1))
|
|
|
|
# Assure there are no prediction records
|
|
load_provider.clear()
|
|
assert len(load_provider) == 0
|
|
|
|
# Execute the method
|
|
load_provider._update_data()
|
|
|
|
# Validate that update_value is called
|
|
assert len(load_provider) > 0
|
|
|
|
|
|
def test_calculate_adjustment(load_provider, measurement_eos):
|
|
"""Test `_calculate_adjustment` for various scenarios."""
|
|
data_year_energy = np.random.rand(365, 2, 24)
|
|
|
|
# Call the method and validate results
|
|
weekday_adjust, weekend_adjust = load_provider._calculate_adjustment(data_year_energy)
|
|
assert weekday_adjust.shape == (24,)
|
|
assert weekend_adjust.shape == (24,)
|
|
|
|
data_year_energy = np.zeros((365, 2, 24))
|
|
weekday_adjust, weekend_adjust = load_provider._calculate_adjustment(data_year_energy)
|
|
|
|
assert weekday_adjust.shape == (24,)
|
|
expected = np.array(
|
|
[
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
100.0,
|
|
]
|
|
)
|
|
np.testing.assert_array_equal(weekday_adjust, expected)
|
|
|
|
assert weekend_adjust.shape == (24,)
|
|
expected = np.array(
|
|
[
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
]
|
|
)
|
|
np.testing.assert_array_equal(weekend_adjust, expected)
|
|
|
|
|
|
def test_load_provider_adjustments_with_mock_data(load_provider):
|
|
"""Test full integration of adjustments with mock data."""
|
|
with patch(
|
|
"akkudoktoreos.prediction.loadakkudoktor.LoadAkkudoktor._calculate_adjustment"
|
|
) as mock_adjust:
|
|
mock_adjust.return_value = (np.zeros(24), np.zeros(24))
|
|
|
|
# Test execution
|
|
load_provider._update_data()
|
|
assert mock_adjust.called
|