update tests

Author: MaxenceGollier

test/runtests.jl

@@ -91,19 +91,9 @@ for (h, h_name) ∈ ((NormL0(λ), "l0"), (NormL1(λ), "l1"), (IndBallL0(10 * com
       x0[p[1:nz]] = sign.(randn(nz))  # initial guess with nz nonzeros (necessary for h = B0)
       args = solver_sym == :LM ? () : (NormLinf(1.0),)
       out = solver(bpdn_nls, h, args..., options, x0 = x0)
-      @test typeof(out.solution) == typeof(bpdn_nls.meta.x0)
-      @test length(out.solution) == bpdn_nls.meta.nvar
-      @test typeof(out.solver_specific[:Fhist]) == typeof(out.solution)
-      @test typeof(out.solver_specific[:Hhist]) == typeof(out.solution)
-      @test typeof(out.solver_specific[:SubsolverCounter]) == Array{Int, 1}
+      @test typeof(out.solution) == typeof(bpdn.meta.x0)
+      @test length(out.solution) == bpdn.meta.nvar
       @test typeof(out.dual_feas) == eltype(out.solution)
-      @test length(out.solver_specific[:Fhist]) == length(out.solver_specific[:Hhist])
-      @test length(out.solver_specific[:Fhist]) == length(out.solver_specific[:SubsolverCounter])
-      @test length(out.solver_specific[:Fhist]) == length(out.solver_specific[:NLSGradHist])
-      @test out.solver_specific[:NLSGradHist][end] ==
-            bpdn_nls.counters.neval_jprod_residual + bpdn_nls.counters.neval_jtprod_residual - 1
-      @test obj(bpdn_nls, out.solution) == out.solver_specific[:Fhist][end]
-      @test h(out.solution) == out.solver_specific[:Hhist][end]
       @test out.status == :first_order
     end
   end

test/test_allocs.jl

@@ -40,7 +40,7 @@ macro wrappedallocs(expr)
 end
 
 # Test non allocating solve!
-@testset "allocs" begin
+@testset "NLP allocs" begin
   for (h, h_name) ∈ ((NormL0(λ), "l0"),)
     for (solver, solver_name) ∈ ((:R2Solver, "R2"), (:R2DHSolver, "R2DH"), (:R2NSolver, "R2N"))
       @testset "$(solver_name)" begin
@@ -59,3 +59,18 @@ end
     end
   end
 end
+
+@testset "NLS allocs" begin
+  for (h, h_name) ∈ ((NormL0(λ), "l0"),)
+    for (solver, solver_name) ∈ ((:LMSolver, "LM"), )
+      @testset "$(solver_name)" begin
+        solver_name == "LM" && continue #FIXME
+        reg_nlp = RegularizedNLPModel(bpdn_nls, h)
+        solver = eval(solver)(reg_nlp)
+        stats = RegularizedExecutionStats(reg_nlp)
+        @test @wrappedallocs(solve!(solver, reg_nlp, stats, σk = 1.0, atol = 1e-6, rtol = 1e-6)) == 0
+        @test stats.status == :first_order
+      end
+    end
+  end
+end

test/test_bounds.jl

@@ -54,16 +54,9 @@ for (h, h_name) ∈ ((NormL0(λ), "l0"), (NormL1(λ), "l1"))
       args = solver_sym == :LM ? () : (NormLinf(1.0),)
       @test has_bounds(bpdn_nls2)
       out = solver(bpdn_nls2, h, args..., options, x0 = x0)
-      @test typeof(out.solution) == typeof(bpdn_nls2.meta.x0)
-      @test length(out.solution) == bpdn_nls2.meta.nvar
-      @test typeof(out.solver_specific[:Fhist]) == typeof(out.solution)
-      @test typeof(out.solver_specific[:Hhist]) == typeof(out.solution)
-      @test typeof(out.solver_specific[:SubsolverCounter]) == Array{Int, 1}
+      @test typeof(out.solution) == typeof(bpdn.meta.x0)
+      @test length(out.solution) == bpdn.meta.nvar
       @test typeof(out.dual_feas) == eltype(out.solution)
-      @test length(out.solver_specific[:Fhist]) == length(out.solver_specific[:Hhist])
-      @test length(out.solver_specific[:Fhist]) == length(out.solver_specific[:SubsolverCounter])
-      @test obj(bpdn_nls2, out.solution) == out.solver_specific[:Fhist][end]
-      @test h(out.solution) == out.solver_specific[:Hhist][end]
       @test out.status == :first_order
     end
   end
@@ -108,20 +101,9 @@ for (h, h_name) ∈ ((NormL0(λ), "l0"),)
     @test has_bounds(bpdn_nls2)
     LM_out =
       LM(bpdn_nls2, h, options, x0 = x0, subsolver = R2DH, subsolver_options = subsolver_options)
-    @test typeof(LM_out.solution) == typeof(bpdn_nls2.meta.x0)
-    @test length(LM_out.solution) == bpdn_nls2.meta.nvar
-    @test typeof(LM_out.solver_specific[:Fhist]) == typeof(LM_out.solution)
-    @test typeof(LM_out.solver_specific[:Hhist]) == typeof(LM_out.solution)
-    @test typeof(LM_out.solver_specific[:SubsolverCounter]) == Array{Int, 1}
-    @test typeof(LM_out.dual_feas) == eltype(LM_out.solution)
-    @test length(LM_out.solver_specific[:Fhist]) == length(LM_out.solver_specific[:Hhist])
-    @test length(LM_out.solver_specific[:Fhist]) ==
-          length(LM_out.solver_specific[:SubsolverCounter])
-    @test length(LM_out.solver_specific[:Fhist]) == length(LM_out.solver_specific[:NLSGradHist])
-    @test LM_out.solver_specific[:NLSGradHist][end] ==
-          bpdn_nls2.counters.neval_jprod_residual + bpdn_nls2.counters.neval_jtprod_residual - 1
-    @test obj(bpdn_nls2, LM_out.solution) == LM_out.solver_specific[:Fhist][end]
-    @test h(LM_out.solution) == LM_out.solver_specific[:Hhist][end]
+    @test typeof(LM_out.solution) == typeof(bpdn.meta.x0)
+    @test length(LM_out.solution) == bpdn.meta.nvar
+    @test typeof(LM_out.dual_feas) == eltype(out.solution)
     @test LM_out.status == :first_order
   end
 end