29 lines
866 B
Python
29 lines
866 B
Python
import pytest
|
|
from energi_bridge.schemas import LocalProblem, SharedVariables, PlanDelta, DualVariables, AuditLog, PrivacyBudget
|
|
from datetime import datetime
|
|
|
|
|
|
def test_local_problem_defaults():
|
|
lp = LocalProblem(id="lp1", domain="solar", assets=["panel1"])
|
|
assert lp.id == "lp1"
|
|
assert lp.domain == "solar"
|
|
assert lp.assets == ["panel1"]
|
|
assert isinstance(lp.constraints, dict)
|
|
|
|
|
|
def test_shared_variables_defaults():
|
|
sv = SharedVariables(version=1)
|
|
assert sv.version == 1
|
|
assert sv.aggregation_method == "mean"
|
|
|
|
|
|
def test_plan_delta_creation():
|
|
pd = PlanDelta(delta={"dispatch": "up"}, timestamp=datetime.utcnow(), author="tester", contract_id="c1")
|
|
assert pd.delta["dispatch"] == "up"
|
|
assert pd.author == "tester"
|
|
|
|
|
|
def test_dual_variables_defaults():
|
|
dv = DualVariables()
|
|
assert dv.lagrange_multipliers == {}
|