18 lines
684 B
Python
18 lines
684 B
Python
import time
|
|
from energiamesh.core import LocalProblem, SharedVariables, PlanDelta, DualVariables, AuditLog
|
|
|
|
|
|
def test_core_dataclasses_simple():
|
|
lp = LocalProblem(site_id="site-1", problem_id="p1", description="test", data={"a": 1})
|
|
sv = SharedVariables(signals={"forecast": 10}, version=1)
|
|
dv = DualVariables(multipliers={"lambda": 0.5}, version=1)
|
|
pd = PlanDelta(delta={"dx": 1}, metadata={"source": "test"}, timestamp=time.time())
|
|
log = AuditLog()
|
|
log.log("created", {"obj": lp.problem_id})
|
|
|
|
assert lp.site_id == "site-1"
|
|
assert sv.version == 1
|
|
assert dv.multipliers["lambda"] == 0.5
|
|
assert "dx" in pd.delta
|
|
assert len(log.entries) == 1
|