add test

Author: lostella

test/problems/test_lasso_small.jl

@@ -138,4 +138,20 @@
 
     end
 
+    @testset "AFBA" begin
+
+        x0 = zeros(T, n)
+        y0 = zeros(T, n)
+
+        solver = ProximalAlgorithms.AFBA{R}(theta=R(1), mu=R(1), tol=R(1e-6))
+        x_afba, y_afba, it_afba = solver(
+            x0, y0, f=f2, g=g, betaQ=opnorm(A)^2,
+        )
+        @test eltype(x_afba) == T
+        @test eltype(y_afba) == T
+        @test norm(x_afba - x_star, Inf) <= 1e-4
+        @test it_afba <= 80
+
+    end
+
 end