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)