13 lines
482 B
Python
13 lines
482 B
Python
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
|