More tests

Author: devmotion

test/kernels/tensorproduct.jl

@@ -12,6 +12,7 @@
     kernel2 = TensorProduct([k1, k2])
 
     @test kernel1.kernels === (k1, k2) === TensorProduct((k1, k2)).kernels
+    @test length(kernel1) == length(kernel2) == 2
 
     @testset "kappa" begin
         for (x, y) in (((v1, u1), (v2, u2)), ([v1, u1], [v2, u2]))
@@ -24,8 +25,8 @@
     end
 
     @testset "kernelmatrix" begin
-        X = rand(2, 10)
-        Y = rand(2, 10)
+        X = rand(rng, 2, 10)
+        Y = rand(rng, 2, 10)
         trueX = kernelmatrix(k1, X[1, :]) .* kernelmatrix(k2, X[2, :])
         trueXY = kernelmatrix(k1, X[1, :], Y[1, :]) .* kernelmatrix(k2, X[2, :], Y[2, :])
         tmp = Matrix{Float64}(undef, 10, 10)
@@ -56,7 +57,7 @@
     end
 
     @testset "kerneldiagmatrix" begin
-        X = rand(2, 10)
+        X = rand(rng, 2, 10)
         trueval = ones(10)
         tmp = Vector{Float64}(undef, 10)
 
@@ -73,4 +74,65 @@
             @test tmp == trueval
         end
     end
+
+    @testset "single kernel" begin
+        kernel = TensorProduct(k1)
+        @test length(kernel) == 1
+
+        @testset "kappa" begin
+            for (x, y) in (((v1,), (v2,)), ([v1], [v2]))
+                val = k1(x[1], y[1])
+
+                @test kernel(x, y) == val
+                @test KernelFunctions.kappa(kernel, x, y) == val
+            end
+        end
+
+        @testset "kernelmatrix" begin
+            X = rand(rng, 1, 10)
+            Y = rand(rng, 1, 10)
+            trueX = kernelmatrix(k1, X)
+            trueXY = kernelmatrix(k1, X, Y)
+            tmp = Matrix{Float64}(undef, 10, 10)
+
+            @test kernelmatrix(kernel, X) == trueX
+            @test kernelmatrix(kernel, X'; obsdim = 1) == trueX
+
+            @test kernelmatrix(kernel, X, Y) == trueXY
+            @test kernelmatrix(kernel, X', Y'; obsdim = 1) == trueXY
+
+            fill!(tmp, 0)
+            kernelmatrix!(tmp, kernel, X)
+            @test tmp == trueX
+
+            fill!(tmp, 0)
+            kernelmatrix!(tmp, kernel, X'; obsdim = 1)
+            @test tmp == trueX
+
+            fill!(tmp, 0)
+            kernelmatrix!(tmp, kernel, X, Y)
+            @test tmp == trueXY
+
+            fill!(tmp, 0)
+            kernelmatrix!(tmp, kernel, X', Y'; obsdim = 1)
+            @test tmp == trueXY
+        end
+
+        @testset "kerneldiagmatrix" begin
+            X = rand(rng, 1, 10)
+            trueval = ones(10)
+            tmp = Vector{Float64}(undef, 10)
+
+            @test kerneldiagmatrix(kernel, X) == trueval
+            @test kerneldiagmatrix(kernel, X'; obsdim = 1) == trueval
+
+            fill!(tmp, 0)
+            kerneldiagmatrix!(tmp, kernel, X)
+            @test tmp == trueval
+
+            fill!(tmp, 0)
+            kerneldiagmatrix!(tmp, kernel, X'; obsdim = 1)
+            @test tmp == trueval
+        end
+    end
 end