def test_admm_with_large_lamduh(N, p, nchunks):
X = da.random.random((N, p), chunks=(N // nchunks, p))
beta = np.random.random(p)
y = make_y(X, beta=np.array(beta), chunks=(N // nchunks, ))
X, y = persist(X, y)
z = admm(X, y, reg=L1, lamduh=1e4, rho=20, max_iter=500)
assert np.allclose(z, np.zeros(p), atol=1e-4)
def test_admm_with_large_lamduh(N, p, nchunks, is_cupy):
X = da.random.random((N, p), chunks=(N // nchunks, p))
beta = np.random.random(p)
y = make_y(X, beta=np.array(beta), chunks=(N // nchunks, ))
if is_cupy:
cupy = pytest.importorskip('cupy')
X, y = to_dask_cupy_array_xy(X, y, cupy)
X, y = persist(X, y)
z = admm(X, y, regularizer=L1(), lamduh=1e5, rho=20, max_iter=500)
assert np.allclose(z, np.zeros(p), atol=1e-4)