21 lines
648 B
Python
21 lines
648 B
Python
import pytest
|
|
from citygrid.core import LocalProblem, SharedVariables, DualVariables, PlanDelta, PrivacyBudget, AuditLog, PolicyBlock
|
|
|
|
|
|
def test_local_problem_dataclass():
|
|
lp = LocalProblem(id="LP1", domain="electric", assets=["b1"], objective={"minimize": "cost"})
|
|
assert lp.id == "LP1"
|
|
assert lp.domain == "electric"
|
|
assert lp.assets == ["b1"]
|
|
|
|
|
|
def test_shared_variables_dataclass():
|
|
sv = SharedVariables(version=1, signals={"load": 42})
|
|
assert sv.version == 1
|
|
assert sv.signals["load"] == 42
|
|
|
|
|
|
def test_dual_variables_dataclass():
|
|
dv = DualVariables(multipliers={"x": 0.1})
|
|
assert dv.multipliers["x"] == 0.1
|