Use ballocated

Author: YingboMa

Project.toml

@@ -13,7 +13,8 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 UnPack = "3a884ed6-31ef-47d7-9d2a-63182c4928ed"
 
 [extras]
+BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test"]
+test = ["BenchmarkTools", "Test"]

test/runtests.jl

@@ -1,5 +1,6 @@
 using NonlinearSolve
 using StaticArrays
+using BenchmarkTools
 using Test
 
 function benchmark_immutable(f, u0)
@@ -28,6 +29,6 @@ sol = benchmark_mutable(f, u0)
 sol = benchmark_scalar(sf, su0)
 @test sol * sol - 2 < 1e-9
 
-@allocated benchmark_immutable(f, u0)
-@allocated benchmark_mutable(f, u0)
-@allocated benchmark_scalar(sf, su0)
+@test_broken (@ballocated benchmark_immutable($f, $u0)) == 0
+@test_broken (@ballocated benchmark_mutable($f, $u0)) == 0
+@test (@ballocated benchmark_scalar($sf, $su0)) == 0