Add tests of pairwise! for ColVecs and RowVecs

Author: devmotion

test/utils.jl

@@ -24,6 +24,12 @@
         DY = ColVecs(Y)
         @test pairwise(SqEuclidean(), DX) ≈ pairwise(SqEuclidean(), X; dims=2)
         @test pairwise(SqEuclidean(), DX, DY) ≈ pairwise(SqEuclidean(), X, Y; dims=2)
+        K = zeros(N, N)
+        pairwise!(K, SqEuclidean(), DX)
+        @test K ≈ pairwise(SqEuclidean(), X; dims=2)
+        K = zeros(N, N + 1)
+        pairwise!(K, SqEuclidean(), DX, DY)
+        @test K ≈ pairwise(SqEuclidean(), X, Y; dims=2)
 
         let
             @test Zygote.pullback(ColVecs, X)[1] == DX
@@ -52,6 +58,12 @@
         DY = RowVecs(Y)
         @test pairwise(SqEuclidean(), DX) ≈ pairwise(SqEuclidean(), X; dims=1)
         @test pairwise(SqEuclidean(), DX, DY) ≈ pairwise(SqEuclidean(), X, Y; dims=1)
+        K = zeros(D, D)
+        pairwise!(K, SqEuclidean(), DX)
+        @test K ≈ pairwise(SqEuclidean(), X; dims=1)
+        K = zeros(D, D + 1)
+        pairwise!(K, SqEuclidean(), DX, DY)
+        @test K ≈ pairwise(SqEuclidean(), X, Y; dims=1)
 
         let
             @test Zygote.pullback(RowVecs, X)[1] == DX