Corrected tests

Author: theogf

test/testAD.jl

@@ -17,13 +17,14 @@ testfunction(k,A) = sum(kernelmatrix(k,A))
 #For debugging
 
 ## Zygote
-# Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),vl)
-# Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A),vl)
-# Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),l)
-# Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A),l)
-
+Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A),vl)
+Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),vl)
+Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),l)
+Zygote.gradient(x->testfunction(SquaredExponentialKernel(x),A),l)
 ## Tracker
-Tracker.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),vl)
+Tracker.gradient(x->testfunction(SquaredExponentialKernel(vl),x,B),A)
+Tracker.gradient(x->testfunction(SquaredExponentialKernel(l),x[:,:]),A)
+# Tracker.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),vl)
 Tracker.gradient(x->testfunction(SquaredExponentialKernel(x),A),vl)
 Tracker.gradient(x->testfunction(SquaredExponentialKernel(x),A,B),l)
 Tracker.gradient(x->testfunction(SquaredExponentialKernel(x),A),l)
@@ -56,42 +57,42 @@ ForwardDiff.gradient(x->testfunction(SquaredExponentialKernel(x[1]),A),[l])
     end
 end
 
-@testset "Tracker AutomaticDifferentation test" begin
+@testset "ForwardDiff AutomaticDifferentation test" begin
     @testset "ARD" begin
         for k in kernels
-            @test_nowarn Tracker.gradient(x->testfunction(k(x),A,B),vl)
-            @test_broken Tracker.gradient(x->testfunction(k(vl),x,B),A)
-            @test_nowarn Tracker.gradient(x->testfunction(k(x),A),vl)
-            @test_broken Tracker.gradient(x->testfunction(k(vl),x),A)
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x),A,B),vl)
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(vl),x,B),A)
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x),A),vl)
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(vl),x),A)
         end
     end
     @testset "ISO" begin
         for k in kernels
-            @test_nowarn Tracker.gradient(x->testfunction(k(x[1]),A,B),[l])
-            @test_broken Tracker.gradient(x->testfunction(k(l),x,B),A)
-            @test_nowarn Tracker.gradient(x->testfunction(k(x[1]),A),[l])
-            @test_broken Tracker.gradient(x->testfunction(k(l),x),A)
-
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x[1]),A,B),[l])
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(l),x,B),A)
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x[1]),A),[l])
+            @test_nowarn ForwardDiff.gradient(x->testfunction(k(l[1]),x),A)
         end
     end
 end
 
 
-@testset "ForwardDiff AutomaticDifferentation test" begin
+@testset "Tracker AutomaticDifferentation test" begin
     @testset "ARD" begin
         for k in kernels
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x),A,B),vl)
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(vl),x,B),A)
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x),A),vl)
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(vl),x),A)
+            @test all(Tracker.gradient(x->testfunction(k(x),A,B),vl)[1] .≈ ForwardDiff.gradient(x->testfunction(k(x),A,B),vl))
+            @test_broken all(Tracker.gradient(x->testfunction(k(vl),x,B),A)[1] .≈ ForwardDiff.gradient(x->testfunction(k(vl),x,B),A))
+            @test all(Tracker.gradient(x->testfunction(k(x),A),vl)[1] .≈  ForwardDiff.gradient(x->testfunction(k(x),A),vl))
+            @test_broken all.(Tracker.gradient(x->testfunction(k(vl),x),A) .≈ ForwardDiff.gradient(x->testfunction(k(vl),x),A))
         end
     end
     @testset "ISO" begin
         for k in kernels
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x[1]),A,B),[l])
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(l),x,B),A)
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(x[1]),A),[l])
-            @test_nowarn ForwardDiff.gradient(x->testfunction(k(l[1]),x),A)
+            @test_nowarn Tracker.gradient(x->testfunction(k(x[1]),A,B),[l])
+            @test_broken Tracker.gradient(x->testfunction(k(l),x,B),A)
+            @test_nowarn Tracker.gradient(x->testfunction(k(x[1]),A),[l])
+            @test_broken Tracker.gradient(x->testfunction(k(l),x),A)
+
         end
     end
 end

test/types.jl

@@ -0,0 +1,3 @@
+## Testing output types from kernel
+
+#TODO