from gridverse.contracts import LocalProblem, ConstraintSet
from gridverse.solver import admm_solve


def test_admm_solve_returns_delta():
    lp = LocalProblem(id="lp-1", description="test", parameters={"p1": 2, "p2": 3})
    cs = ConstraintSet(name="mesh", constraints={"energy": 1})
    delta = admm_solve(lp, {}, cs)
    assert delta.id.startswith("pd-")
    assert isinstance(delta.updates, dict)
    assert delta.updates.get("p1") == 2
    assert delta.updates.get("p2") == 3