def test_l2_proximal_operator(beta, expected):
npt.assert_array_equal(regs.L2().proximal_operator(beta, 1), expected)
def test_l2_hessian(beta):
npt.assert_array_equal(regs.L2().hessian(beta), np.eye(len(beta)))
def test_l2_gradient(beta):
npt.assert_array_equal(regs.L2().gradient(beta), beta)
def test_l2_function(beta, expected):
assert regs.L2().f(beta) == expected
def test_elastic_net_gradient_zero_weight_is_l2():
beta = np.array([1, 2, 3])
npt.assert_array_equal(
regs.ElasticNet(weight=0).gradient(beta),
regs.L2().gradient(beta))
def test_elastic_net_function_zero_weight_is_l2():
beta = np.array([1, 2, 3])
assert regs.ElasticNet(weight=0).f(beta) == regs.L2().f(beta)