add VoA and StructArray compatibility

Author: jlchan

lib/OrdinaryDiffEqLowStorageRK/test/ode_low_storage_rk_tests.jl

@@ -1605,3 +1605,26 @@ end
         save_start = false, alias = ODEAliasSpecifier(alias_u0 = true))
     @test sol_old[end] ≈ sol_new[end]
 end
+
+@testset "VectorOfArray/StructArray compatibility" begin
+    using RecursiveArrayTools, StaticArrays, StructArrays
+
+    function rhs!(du_voa, u_voa, p, t)
+        du = parent(du_voa)
+        u = parent(u_voa)
+        du .= u
+    end
+
+    # StructArray storage
+    u = StructArray{SVector{1, Float64}}(ntuple(_ -> [1.0, 2.0], 1))
+    ode = ODEProblem(rhs!, VectorOfArray(u), (0, 0.7))
+    sol_SA = solve(ode, RDPK3SpFSAL35())
+
+    # Vector{<:SVector} storage
+    u = SVector{1, Float64}.([1.0, 2.0])
+    ode = ODEProblem(rhs!, VectorOfArray(u), (0, 0.7))
+    sol_SV = solve(ode, RDPK3SpFSAL35())
+
+    @test sol.SA ≈ sol_SV
+    @test sol_SV.stats.naccept == sol_SA.stats.naccept
+end

lib/OrdinaryDiffEqSSPRK/test/ode_ssprk_tests.jl

@@ -491,3 +491,26 @@ end
 sol = solve(test_problem_ssp_long, alg, dt = OrdinaryDiffEqSSPRK.ssp_coefficient(alg),
     dense = false)
 @test all(sol.u .>= 0)
+
+@testset "VectorOfArray/StructArray compatibility" begin
+    using RecursiveArrayTools, StaticArrays, StructArrays
+
+    function rhs!(du_voa, u_voa, p, t)
+        du = parent(du_voa)
+        u = parent(u_voa)
+        du .= u
+    end
+
+    # StructArray storage
+    u = StructArray{SVector{1, Float64}}(ntuple(_ -> [1.0, 2.0], 1))
+    ode = ODEProblem(rhs!, VectorOfArray(u), (0, 0.7))
+    sol_SA = solve(ode, SSPRK43())
+
+    # Vector{<:SVector} storage
+    u = SVector{1, Float64}.([1.0, 2.0])
+    ode = ODEProblem(rhs!, VectorOfArray(u), (0, 0.7))
+    sol_SV = solve(ode, SSPRK43())
+
+    @test sol.SA ≈ sol_SV
+    @test sol_SV.stats.naccept == sol_SA.stats.naccept
+end