Add tests

Author: gdalle

DifferentiationInterface/test/Core/Internals/linalg.jl

@@ -1,6 +1,7 @@
-using DifferentiationInterface: recursive_similar, get_pattern
+using DifferentiationInterface: recursive_similar, get_pattern, arroftup_to_tupofarr
 using SparseArrays
 using Test
+using JLArrays, ComponentArrays
 
 @testset "Recursive similar" begin
     @test recursive_similar(ones(Int, 2), Float32) isa Vector{Float32}
@@ -16,3 +17,19 @@ end
     @test_broken get_pattern(D) == Diagonal(trues(10))
     @test get_pattern(sparse(D)) == Diagonal(trues(10))
 end
+
+@testset "Wrong-mode array conversion" begin
+    x = [1.0, 3.0, 5.0]
+    xt = [(1.0, 2.0), (3.0, 4.0), (5.0, 6.0)]
+    y = ComponentVector(a = [1.0, 3.0], b = [5.0])
+    yt = ComponentVector(a = [(1.0, 2.0), (3.0, 4.0)], b = [(5.0, 6.0)])
+    z = jl([1.0, 3.0, 5.0])
+    zt = jl([(1.0, 2.0), (3.0, 4.0), (5.0, 6.0)])
+    @test arroftup_to_tupofarr((1.0, 2.0), 1.0) == (1.0, 2.0)
+    @test arroftup_to_tupofarr(xt, x) == ([1.0, 3.0, 5.0], [2.0, 4.0, 6.0])
+    @test arroftup_to_tupofarr(yt, y) == (ComponentVector(a = [1.0, 3.0], b = [5.0]), ComponentVector(a = [2.0, 4.0], b = [6.0]))
+    @test arroftup_to_tupofarr(zt, z) == (jl([1.0, 3.0, 5.0]), jl([2.0, 4.0, 6.0]))
+    @test arroftup_to_tupofarr(xt, x)[1] isa Vector
+    @test arroftup_to_tupofarr(yt, y)[1] isa ComponentVector
+    @test arroftup_to_tupofarr(zt, z)[1] isa JLVector
+end

DifferentiationInterface/test/Core/SimpleFiniteDiff/test.jl

@@ -169,3 +169,8 @@ end
         logging = LOGGING,
     )
 end;
+
+@testset "Array format preservation in wrong mode" begin
+    @test gradient(sum, AutoSimpleFiniteDiff(), jl(ones(2))) isa JLVector
+    @test derivative(t -> jl(fill(t, 2)), AutoReverseFromPrimitive(AutoSimpleFiniteDiff()), 1.0) isa JLVector
+end