Fix static arrays

Author: gdalle

DifferentiationInterfaceTest/ext/DifferentiationInterfaceTestStaticArraysExt/DifferentiationInterfaceTestStaticArraysExt.jl

@@ -3,7 +3,7 @@ module DifferentiationInterfaceTestStaticArraysExt
 import DifferentiationInterface as DI
 import DifferentiationInterfaceTest as DIT
 using SparseArrays: SparseArrays, SparseMatrixCSC, nnz, spdiagm
-using StaticArrays: MArray, MMatrix, MVector, SArray, SMatrix, SVector
+using StaticArrays: StaticArray, MArray, MMatrix, MVector, SArray, SMatrix, SVector
 
 static_num_to_vec(x::Number) = sin.(SVector(1, 2) .* x)
 static_num_to_mat(x::Number) = hcat(static_num_to_vec(x), static_num_to_vec(3x))
@@ -37,13 +37,19 @@ mystatic(::Nothing) = nothing
 
 function mystatic(scen::DIT.Scenario{op,pl_op,pl_fun}) where {op,pl_op,pl_fun}
     (; f, x, y, t, contexts, prep_args, res1, res2, name) = scen
+    new_prep_args = (;
+        x=mystatic(prep_args.x), contexts=map(mystatic, prep_args.contexts), t=mystatic(t)
+    )
+    if pl_fun == :in
+        new_prep_args = (; new_prep_args..., y=mymutablestatic(prep_args.y))
+    end
     return DIT.Scenario{op,pl_op,pl_fun}(;
         f=mystatic(f),
         x=mystatic(x),
         y=pl_fun == :in ? mymutablestatic(y) : mystatic(y),
         t=mystatic(t),
         contexts=mystatic(contexts),
-        prep_args=map(mystatic, prep_args),
+        prep_args=new_prep_args,
         res1=mystatic(res1),
         res2=mystatic(res2),
         name=name,