add zygote tests

Author: lkdvos

test/chainrules.jl

@@ -287,6 +287,10 @@ end
         config, last ∘ eig_full, A;
         output_tangent = ΔV, atol = atol, rtol = rtol, rrule_f = rrule_via_ad, check_inferred = false
     )
+    test_rrule(
+        config, eig_vals, A;
+        output_tangent = diagview(ΔD), atol, rtol, rrule_f = rrule_via_ad, check_inferred = false
+    )
 end
 
 @timedtestset "EIGH AD Rules with eltype $T" for T in (Float64, ComplexF64, Float32)
@@ -365,6 +369,10 @@ end
         config, last ∘ eigh_full ∘ Matrix ∘ Hermitian, A;
         output_tangent = ΔV, atol = atol, rtol = rtol, rrule_f = rrule_via_ad, check_inferred = false
     )
+    test_rrule(
+        config, eigh_vals ∘ Matrix ∘ Hermitian, A;
+        output_tangent = diagview(ΔD), atol, rtol, rrule_f = rrule_via_ad, check_inferred = false
+    )
     eigh_trunc2(A; kwargs...) = eigh_trunc(Matrix(Hermitian(A)); kwargs...)
     for r in 1:4:m
         trunc = truncrank(r; by = real)
@@ -459,6 +467,10 @@ end
             output_tangent = (ΔU, ΔS2, ΔVᴴ), atol = atol, rtol = rtol,
             rrule_f = rrule_via_ad, check_inferred = false
         )
+        test_rrule(
+            config, svd_vals, A;
+            output_tangent = diagview(ΔS), atol, rtol, rrule_f = rrule_via_ad, check_inferred = false
+        )
         for r in 1:4:minmn
             trunc = truncrank(r)
             ind = MatrixAlgebraKit.findtruncated(diagview(S), trunc)