Merge pull request #94 from ChrisRackauckas-Claude/interface-check-20260101-030232

Add JLArray tests for GPU-like array interface compliance

Author: ChrisRackauckas

Project.toml

@@ -14,17 +14,20 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 
 [compat]
 DiffEqBase = "6.122"
+JLArrays = "0.1, 0.2, 0.3"
 MuladdMacro = "0.2"
+OrdinaryDiffEq = "6"
 Parameters = "0.12"
 RecursiveArrayTools = "2, 3"
 Reexport = "0.2, 1.0"
 StaticArrays = "0.10, 0.11, 0.12, 1.0"
 julia = "1.6"
 
 [extras]
+JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["OrdinaryDiffEq", "SafeTestsets", "Test"]
+test = ["JLArrays", "OrdinaryDiffEq", "SafeTestsets", "Test"]

test/interface_tests.jl

@@ -1,4 +1,4 @@
-using SimpleDiffEq, StaticArrays, Test
+using SimpleDiffEq, StaticArrays, JLArrays, Test
 
 # Test interface compatibility with different number types
 
@@ -84,3 +84,50 @@ end
     sol = solve(prob, GPUSimpleATsit5(), dt = dt)
     @test eltype(sol.u[end]) == BigFloat
 end
+
+# Test JLArray support for GPU-like array interface compliance
+# JLArrays provide a GPU-like array that catches interface violations
+# such as improper scalar indexing or type hardcoding
+
+@testset "JLArray support (OOP)" begin
+    u0 = JLArray([1.0, 2.0, 3.0])
+    tspan = (0.0, 1.0)
+    dt = 0.01
+    prob = ODEProblem{false}(decay, u0, tspan, nothing)
+
+    # Fixed step solvers
+    sol = solve(prob, SimpleEuler(), dt = dt)
+    @test sol.u[end] isa JLArray
+
+    sol = solve(prob, SimpleRK4(), dt = dt)
+    @test sol.u[end] isa JLArray
+
+    sol = solve(prob, SimpleTsit5(), dt = dt)
+    @test sol.u[end] isa JLArray
+
+    # Adaptive solver
+    sol = solve(prob, SimpleATsit5(), dt = dt)
+    @test sol.u[end] isa JLArray
+
+    # GPU-optimized solvers
+    sol = solve(prob, GPUSimpleTsit5(), dt = dt)
+    @test sol.u[end] isa JLArray
+
+    sol = solve(prob, GPUSimpleATsit5(), dt = dt)
+    @test sol.u[end] isa JLArray
+end
+
+@testset "JLArray scalar support (OOP)" begin
+    # Test with scalar wrapped in JLArray-compatible manner
+    # GPU solvers should handle scalar problems correctly
+    u0 = 1.0
+    tspan = (0.0, 1.0)
+    dt = 0.01
+    prob = ODEProblem{false}(decay, u0, tspan, nothing)
+
+    sol = solve(prob, GPUSimpleTsit5(), dt = dt)
+    @test eltype(sol.u) == Float64
+
+    sol = solve(prob, GPUSimpleATsit5(), dt = dt)
+    @test eltype(sol.u) == Float64
+end