change to alloccheck.jl

Author: ChrisRackauckas

Project.toml

@@ -147,6 +147,7 @@ julia = "1.10"
 
 [extras]
 AlgebraicMultigrid = "2169fc97-5a83-5252-b627-83903c6c433c"
+AllocCheck = "9b6a8646-10ed-4001-bbdc-1d2f46dfbb1a"
 Calculus = "49dc2e85-a5d0-5ad3-a950-438e2897f1b9"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 DiffEqCallbacks = "459566f4-90b8-5000-8ac3-15dfb0a30def"
@@ -171,4 +172,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [targets]
-test = ["Calculus", "ComponentArrays", "Symbolics", "AlgebraicMultigrid", "IncompleteLU", "DiffEqCallbacks", "DiffEqDevTools", "ODEProblemLibrary", "ElasticArrays", "InteractiveUtils", "ParameterizedFunctions", "PoissonRandom", "Printf", "Random", "ReverseDiff", "SafeTestsets", "SparseArrays", "Statistics", "Test", "Unitful", "ModelingToolkit", "Pkg", "NLsolve"]
+test = ["Calculus", "AllocCheck", "ComponentArrays", "Symbolics", "AlgebraicMultigrid", "IncompleteLU", "DiffEqCallbacks", "DiffEqDevTools", "ODEProblemLibrary", "ElasticArrays", "InteractiveUtils", "ParameterizedFunctions", "PoissonRandom", "Printf", "Random", "ReverseDiff", "SafeTestsets", "SparseArrays", "Statistics", "Test", "Unitful", "ModelingToolkit", "Pkg", "NLsolve"]

test/integrators/callback_allocation_tests.jl

@@ -1,5 +1,5 @@
 using OrdinaryDiffEq, Test
-using OrdinaryDiffEqCore
+using OrdinaryDiffEqCore, AllocCheck
 
 # Setup a simple ODE problem with several callbacks (to test LLVM code gen)
 # We will manually trigger the first callback and check its allocations.
@@ -39,11 +39,9 @@ integrator = init(
 # call handle callbacks
 step!(integrator, 0.1, true)
 
-if VERSION >= v"1.7"
-    function handle_allocs(integrator)
-        integrator.u[1] = 0.4
-        @allocations OrdinaryDiffEqCore.handle_callbacks!(integrator)
-    end
-    handle_allocs(integrator)
-    @test handle_allocs(integrator) == 0
-end
+function handle_allocs(integrator)
+    integrator.u[1] = 0.4
+    OrdinaryDiffEqCore.handle_callbacks!(integrator)
+end
+handle_allocs(integrator)
+@check_allocs handle_allocs(integrator)