import numpy as np
from gridforge.solver import admm_lite
def test_admm_lite_runs():
A = np.array([[3.0, 1.0], [1.0, 2.0]])
b = np.array([1.0, 2.0])
x = admm_lite(A, b, max_iter=10, rho=1.0)
assert x.shape[0] == A.shape[1]