add test for successful return code

SciML · Feb 17, 2023 · acecc4a · acecc4a
1 parent 2d1beff
commit acecc4a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/SSA_stepper.jl b/src/SSA_stepper.jl
@@ -126,9 +126,9 @@ function DiffEqBase.solve!(integrator::SSAIntegrator)
     end
 
     DiffEqBase.finalize!(integrator.opts.callback, integrator.u, integrator.t, integrator)
-    
+
     if integrator.sol.retcode === ReturnCode.Default
-      integrator.sol = DiffEqBase.solution_new_retcode(integrator.sol, ReturnCode.Success)
+        integrator.sol = DiffEqBase.solution_new_retcode(integrator.sol, ReturnCode.Success)
     end
 end
 

diff --git a/test/ssa_tests.jl b/test/ssa_tests.jl
@@ -1,4 +1,4 @@
-using JumpProcesses, DiffEqBase
+using JumpProcesses, DiffEqBase, SciMLBase
 using Test
 using StableRNGs
 rng = StableRNG(12345)
@@ -25,6 +25,7 @@ integrator.u[1]
 # test different saving behaviors
 
 sol = solve(jump_prob, SSAStepper())
+@test SciMLBase.successful_retcode(sol)
 @test sol.t[begin] == 0.0
 @test sol.t[end] == 3.0