Spread tests for all transforms

Author: theogf

test/transform/ardtransform.jl

@@ -41,4 +41,5 @@
     @test_throws DimensionMismatch map(t, ColVecs(randn(rng, D + 1, 3)))
 
     @test repr(t) == "ARD Transform (dims: $D)"
+    test_ADs(x->transform(SEKernel(), exp.(x)), randn(rng, 3))
 end

test/transform/chaintransform.jl

@@ -22,8 +22,5 @@
 
     # Verify printing works as expected.
     @test repr(tp ∘ tf) == "Chain of 2 transforms:\n\t - $(tf) |> $(tp)"
+    test_ADs(x->transform(SEKernel(), ScaleTransform(exp(x[1])) ∘ ARDTransform(exp.(x[2:4]))), randn(rng, 4))
 end
-
-
-Base.:∘(t::Transform, tc::ChainTransform) = ChainTransform(vcat(tc.transforms, t))
-Base.:∘(tc::ChainTransform, t::Transform) = ChainTransform(vcat(t, tc.transforms))

test/transform/functiontransform.jl

@@ -26,4 +26,7 @@
     end
 
     @test repr(FunctionTransform(sin)) == "Function Transform: $(sin)"
+    f(a, x) = sin.(a .* x)
+    test_ADs(x->transform(SEKernel(), FunctionTransform(y->f(x, y))), randn(rng, 3), ADs = [:ForwardDiff, :ReverseDiff])
+    @test_broken "Zygote is failing"
 end

test/transform/lineartransform.jl

@@ -41,4 +41,5 @@
     @test_throws DimensionMismatch map(t, ColVecs(randn(rng, Din + 1, Dout)))
 
     @test repr(t) == "Linear transform (size(A) = ($Dout, $Din))"
+    test_ADs(x->transform(SEKernel(), LinearTransform(x)), randn(rng, 3, 3))
 end

test/transform/scaletransform.jl

@@ -18,4 +18,5 @@
     @test t.s == [s2]
     @test isequal(ScaleTransform(s), ScaleTransform(s))
     @test repr(t) == "Scale Transform (s = $(s2))"
+    test_ADs(x->transform(SEKernel(), exp(x[1])), randn(rng, 1))
 end

test/transform/selecttransform.jl

@@ -18,4 +18,5 @@
     @test t.select == select2
 
     @test repr(t) == "Select Transform (dims: $(select2))"
+    test_ADs(()->transform(SEKernel(), SelectTransform([1,2])))
 end

test/transform/transform.jl

@@ -7,4 +7,5 @@
         @test IdentityTransform()(x) == x
         @test map(IdentityTransform(), x) == x
     end
+    test_ADs(()->transform(SEKernel(), IdentityTransform()))
 end