Added tests for inplace RODESolution interpolation

Author: Arthur Amorim

Project.toml

@@ -1,7 +1,7 @@
 name = "StochasticDiffEq"
 uuid = "789caeaf-c7a9-5a7d-9973-96adeb23e2a0"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "6.61.2a"
+version = "6.61.2b"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

test/inplace_interpolation.jl

@@ -0,0 +1,38 @@
+using StochasticDiffEq, Test, Random
+import SDEProblemLibrary: prob_sde_linear, prob_sde_2Dlinear
+Random.seed!(100)
+
+vecarrzero(m::Integer, n) = map(t->zeros(n), 1:m)
+
+tt = 0:0.05:1
+ntt = length(tt)
+out_VF = zeros(ntt)                                     # Vector{Float64}
+out_VVF_1 = vecarrzero(ntt,1)                           # Vector{Vector{Float64}}
+out_VVF_2 = vecarrzero(ntt,2)                           # Vector{Vector{Float64}}
+out_VMF = vecarrzero(ntt, size(prob_sde_2Dlinear.u0))   # Vector{Matrix{Float64}}
+
+
+@testset verbose=true "SDESolution interpolation" begin
+    sol_SDE = solve(prob_sde_linear, SRIW1(); dt=1//2^4)
+    sol_SDE_2D = solve(prob_sde_2Dlinear, SRIW1(); dt=1//2^4)
+
+    sol_SDE_interp = sol_SDE(tt)
+    sol_SDE_2D_interp = sol_SDE_2D(tt)
+
+    @testset "1D" begin
+        @test_throws MethodError sol_SDE(out_VF, tt; idxs=1:1)
+        @test sol_SDE(out_VF, tt) isa Vector{Float64}
+        @test sol_SDE(out_VVF_1,tt) isa Vector{Vector{Float64}}
+        @test sol_SDE_interp.u == out_VF
+    end
+
+    @testset "2D" begin
+        @test_throws MethodError sol_SDE_2D(out_VF, tt; idxs=3:3)
+        @test sol_SDE_2D(out_VF, tt; idxs=3) isa Vector{Float64}
+        @test sol_SDE_2D(out_VVF_1, tt; idxs=3) isa Vector{Vector{Float64}}
+        @test sol_SDE_2D(out_VVF_1, tt; idxs=3:3) isa Vector{Vector{Float64}}
+        @test sol_SDE_2D(out_VVF_2, tt; idxs=2:3) isa Vector{Vector{Float64}}
+        @test sol_SDE_2D(out_VMF, tt) isa Vector{Matrix{Float64}}
+        @test sol_SDE_2D_interp.u == out_VMF
+    end
+end

test/runtests.jl

@@ -40,6 +40,7 @@ const is_APPVEYOR = Sys.iswindows() && haskey(ENV,"APPVEYOR")
     @time @safetestset "Scalar Tests" begin include("scalar_noise.jl") end
     @time @safetestset "Stiffness Detection Test" begin include("stiffness_detection_test.jl") end
     @time @safetestset "Adaptive SDE Linear Tests" begin include("adaptive/sde_linearadaptive_tests.jl") end
+    @time @safetestset "Inplace RODESolution Interpolation Tests" begin include("inplace_interpolation.jl") end
   end
 
   if GROUP == "All" || GROUP == "Interface3"