Add allocations tests (#91)

* Add allocations tests

* add tests with new QN solvers

Author: tmigot

Project.toml

@@ -17,7 +17,8 @@ NLPModels = "0.19"
 julia = "^1.6.0"
 
 [extras]
+NLPModelsTest = "7998695d-6960-4d3a-85c4-e1bceb8cd856"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test"]
+test = ["NLPModelsTest", "Test"]

test/allocs_test.jl

@@ -0,0 +1,53 @@
+@testset "Check allocations for model modifiers of NLP" begin
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> LBFGSModel(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> LSR1Model(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> DiagonalQNModel(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> SpectralGradientModel(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> SlackModel(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  problems = setdiff(NLPModelsTest.nlp_problems, ["BROWNDEN", "HS5"])
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess]),
+    map(x -> FeasibilityResidual(eval(Symbol(x))()), problems),
+  )
+end
+
+@testset "Check allocations for model modifiers of NLS" begin
+  problems = NLPModelsTest.nls_problems
+  # `obj` and `grad` of an NLS model allocates a vector of size `nequ`.
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [obj, grad, hess]),
+    map(x -> LBFGSModel(eval(Symbol(x))()), problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [obj, grad, hess]),
+    map(x -> LSR1Model(eval(Symbol(x))()), problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [obj, grad, hess]),
+    map(x -> SlackModel(eval(Symbol(x))()), NLPModelsTest.nlp_problems),
+  )
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [obj, grad, hess, hess_residual]),
+    map(x -> FeasibilityResidual(eval(Symbol(x))(), name = x * "feas"), setdiff(problems, ["MGH01"])),
+  )
+  # jtprod! https://github.com/JuliaSmoothOptimizers/NLPModelsModifiers.jl/issues/77
+  map(
+    nlp -> NLPModelsTest.test_zero_allocations(nlp, linear_api = true, exclude = [hess, jtprod, jac_op]),
+    map(x -> FeasibilityFormNLS(eval(Symbol(x))()), problems),
+  )
+end

test/runtests.jl

@@ -10,6 +10,12 @@ include("nls/feasibility-form-nls.jl")
 include("nls/feasibility-residual.jl")
 include("nls/slack-model.jl")
 
+using NLPModelsTest
+
+if (v"1.7" <= VERSION)
+  include("allocs_test.jl")
+end
+
 # include("test_feasibility_form_nls.jl")
 # include("test_feasibility_nls_model.jl")
 # include("test_qn_model.jl")