test: add tests for zygote rules with matrix inputs

SciML · Jan 21, 2025 · 977f085 · 977f085
1 parent ec804ba
commit 977f085
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/zygote_tests.jl b/test/zygote_tests.jl
@@ -51,12 +51,18 @@ end
     t = collect(1.0:10.0)
     test_zygote(
         LinearInterpolation, u, t; name = "Linear Interpolation")
+    u2 = Matrix(hcat(u, u)')
+    test_zygote(
+        LinearInterpolation, u2, t; name = "Linear Interpolation with matrix input")
 end
 
 @testset "Quadratic Interpolation" begin
     u = [1.0, 4.0, 9.0, 16.0]
     t = [1.0, 2.0, 3.0, 4.0]
     test_zygote(QuadraticInterpolation, u, t; name = "Quadratic Interpolation")
+    u2 = Matrix(hcat(u, u)')
+    test_zygote(
+        QuadraticInterpolation, u2, t; name = "Quadratic Interpolation with matrix input")
 end
 
 @testset "Constant Interpolation" begin