mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2025-04-17 07:55:15 +00:00
aa334d0b61
* Update utilities in utils submodule. * Add base configuration modules. * Add server base configuration modules. * Add devices base configuration modules. * Add optimization base configuration modules. * Add utils base configuration modules. * Add prediction abstract and base classes plus tests. * Add PV forecast to prediction submodule. The PV forecast modules are adapted from the class_pvforecast module and replace it. * Add weather forecast to prediction submodule. The modules provide classes and methods to retrieve, manage, and process weather forecast data from various sources. Includes are structured representations of weather data and utilities for fetching forecasts for specific locations and time ranges. BrightSky and ClearOutside are currently supported. * Add electricity price forecast to prediction submodule. * Adapt fastapi server to base config and add fasthtml server. * Add ems to core submodule. * Adapt genetic to config. * Adapt visualize to config. * Adapt common test fixtures to config. * Add load forecast to prediction submodule. * Add core abstract and base classes. * Adapt single test optimization to config. * Adapt devices to config. Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
from typing import List, Literal, Optional, no_type_check
|
|
|
|
import pytest
|
|
from pydantic import Field, ValidationError
|
|
|
|
from akkudoktoreos.config.configabc import SettingsBaseModel
|
|
|
|
|
|
class SettingsModel(SettingsBaseModel):
|
|
name: str = "Default Name"
|
|
age: int = 18
|
|
tags: List[str] = Field(default_factory=list)
|
|
readonly_field: Literal["ReadOnly"] = "ReadOnly" # Use Literal instead of const
|
|
|
|
|
|
def test_reset_to_defaults():
|
|
"""Test resetting to default values."""
|
|
instance = SettingsModel(name="Custom Name", age=25, tags=["tag1", "tag2"])
|
|
|
|
# Modify the instance
|
|
instance.name = "Modified Name"
|
|
instance.age = 30
|
|
instance.tags.append("tag3")
|
|
|
|
# Ensure the instance is modified
|
|
assert instance.name == "Modified Name"
|
|
assert instance.age == 30
|
|
assert instance.tags == ["tag1", "tag2", "tag3"]
|
|
|
|
# Reset to defaults
|
|
instance.reset_to_defaults()
|
|
|
|
# Verify default values
|
|
assert instance.name == "Default Name"
|
|
assert instance.age == 18
|
|
assert instance.tags == []
|
|
assert instance.readonly_field == "ReadOnly"
|
|
|
|
|
|
@no_type_check
|
|
def test_reset_to_defaults_readonly_field():
|
|
"""Ensure read-only fields remain unchanged."""
|
|
instance = SettingsModel()
|
|
|
|
# Attempt to modify readonly_field (should raise an error)
|
|
with pytest.raises(ValidationError):
|
|
instance.readonly_field = "New Value"
|
|
|
|
# Reset to defaults
|
|
instance.reset_to_defaults()
|
|
|
|
# Ensure readonly_field is still at its default value
|
|
assert instance.readonly_field == "ReadOnly"
|
|
|
|
|
|
def test_reset_to_defaults_with_default_factory():
|
|
"""Test reset with fields having default_factory."""
|
|
|
|
class FactoryModel(SettingsBaseModel):
|
|
items: List[int] = Field(default_factory=lambda: [1, 2, 3])
|
|
value: Optional[int] = None
|
|
|
|
instance = FactoryModel(items=[4, 5, 6], value=10)
|
|
|
|
# Ensure instance has custom values
|
|
assert instance.items == [4, 5, 6]
|
|
assert instance.value == 10
|
|
|
|
# Reset to defaults
|
|
instance.reset_to_defaults()
|
|
|
|
# Verify reset values
|
|
assert instance.items == [1, 2, 3]
|
|
assert instance.value is None
|
|
|
|
|
|
@no_type_check
|
|
def test_reset_to_defaults_error_handling():
|
|
"""Ensure reset_to_defaults skips fields that cannot be set."""
|
|
|
|
class ReadOnlyModel(SettingsBaseModel):
|
|
readonly_field: Literal["ReadOnly"] = "ReadOnly"
|
|
|
|
instance = ReadOnlyModel()
|
|
|
|
# Attempt to modify readonly_field (should raise an error)
|
|
with pytest.raises(ValidationError):
|
|
instance.readonly_field = "New Value"
|
|
|
|
# Reset to defaults
|
|
instance.reset_to_defaults()
|
|
|
|
# Ensure readonly_field is unaffected
|
|
assert instance.readonly_field == "ReadOnly"
|