mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-19 08:55:15 +00:00
929d7e0a75
Update test_class_optimize.py added Skipped per default Fix optimization test Fix optimization test
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from akkudoktoreos.class_optimize import optimization_problem
|
|
|
|
DIR_TESTDATA = Path(__file__).parent / "testdata"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"fn_in, fn_out", [("optimize_input_1.json", "optimize_result_1.json")]
|
|
)
|
|
@pytest.mark.skip(reason="Expensive - Skipped per default")
|
|
def test_optimize(fn_in, fn_out):
|
|
# Load input and output data
|
|
with open(DIR_TESTDATA / fn_in, "r") as f_in:
|
|
input_data = json.load(f_in)
|
|
|
|
with open(DIR_TESTDATA / fn_out, "r") as f_out:
|
|
expected_output_data = json.load(f_out)
|
|
|
|
opt_class = optimization_problem(
|
|
prediction_hours=48, strafe=10, optimization_hours=24, fixed_seed=42
|
|
)
|
|
start_hour = 10
|
|
|
|
# Call the optimization function
|
|
ergebnis = opt_class.optimierung_ems(parameter=input_data, start_hour=start_hour)
|
|
|
|
# Assert that the output contains all expected entries.
|
|
# This does not assert that the optimization always gives the same result!
|
|
# Reproducibility and mathematical accuracy should be tested on the level of individual components.
|
|
assert set(ergebnis) == set(expected_output_data)
|