add unit tests for single-point posterior evals

soldasim · Nov 5, 2024 · dd72220 · dd72220
1 parent 45b1602
commit dd72220
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 3 deletions.
diff --git a/test/unit/test/models/gaussian_process.jl b/test/unit/test/models/gaussian_process.jl
@@ -131,6 +131,11 @@ end
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1,:], out([1., 1.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2,:], out([2., 2.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3,:], out([3., 3.])[2] .^ 2; atol=1e-8) |> all,
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractMatrix{<:Real}, <:AbstractArray{<:Real, 3}},
+            size(out([1.;1.;;])[1]) == (1, 2),
+            size(out([1.;1.;;])[2]) == (1, 1, 2),
         )
     end
 end
@@ -170,14 +175,19 @@ end
 
             # matrix
             out([1.;1.;; 2.;2.;; 3.;3.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
-            length(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == 3,
+            size(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == (3,),
             size(out([1.;1.;; 2.;2.;; 3.;3.;;])[2]) == (3, 3),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][1], out([1., 1.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][2], out([2., 2.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][3], out([3., 3.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1], out([1., 1.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2], out([2., 2.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3], out([3., 3.])[2] ^ 2; atol=1e-8),
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
+            size(out([1.;1.;;])[1]) == (1,),
+            size(out([1.;1.;;])[2]) == (1, 1),
         )
     end
 end

diff --git a/test/unit/test/models/parametric.jl b/test/unit/test/models/parametric.jl
@@ -216,6 +216,11 @@ end
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1,:], out([1., 1.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2,:], out([2., 2.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3,:], out([3., 3.])[2] .^ 2; atol=1e-8) |> all,
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractMatrix{<:Real}, <:AbstractArray{<:Real, 3}},
+            size(out([1.;1.;;])[1]) == (1, 2),
+            size(out([1.;1.;;])[2]) == (1, 1, 2),
         )
     end
 end
@@ -273,14 +278,19 @@ end
 
             # matrix
             out([1.;1.;; 2.;2.;; 3.;3.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
-            length(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == 3,
+            size(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == (3,),
             size(out([1.;1.;; 2.;2.;; 3.;3.;;])[2]) == (3, 3),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][1], out([1., 1.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][2], out([2., 2.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][3], out([3., 3.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1], out([1., 1.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2], out([2., 2.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3], out([3., 3.])[2] ^ 2; atol=1e-8),
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
+            size(out([1.;1.;;])[1]) == (1,),
+            size(out([1.;1.;;])[2]) == (1, 1),
         )
     end
 end

diff --git a/test/unit/test/models/semiparametric.jl b/test/unit/test/models/semiparametric.jl
@@ -48,6 +48,11 @@
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1,:], out([1., 1.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2,:], out([2., 2.])[2] .^ 2; atol=1e-8) |> all,
             isapprox.(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3,:], out([3., 3.])[2] .^ 2; atol=1e-8) |> all,
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractMatrix{<:Real}, <:AbstractArray{<:Real, 3}},
+            size(out([1.;1.;;])[1]) == (1, 2),
+            size(out([1.;1.;;])[2]) == (1, 1, 2),
         )
     end
 end
@@ -92,14 +97,19 @@ end
 
             # matrix
             out([1.;1.;; 2.;2.;; 3.;3.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
-            length(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == 3,
+            size(out([1.;1.;; 2.;2.;; 3.;3.;;])[1]) == (3,),
             size(out([1.;1.;; 2.;2.;; 3.;3.;;])[2]) == (3, 3),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][1], out([1., 1.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][2], out([2., 2.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[1][3], out([3., 3.])[1]; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][1,1], out([1., 1.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][2,2], out([2., 2.])[2] ^ 2; atol=1e-8),
             isapprox(out([1.;1.;; 2.;2.;; 3.;3.;;])[2][3,3], out([3., 3.])[2] ^ 2; atol=1e-8),
+
+            # single-element matrix
+            out([1.;1.;;]) isa Tuple{<:AbstractVector{<:Real}, <:AbstractMatrix{<:Real}},
+            size(out([1.;1.;;])[1]) == (1,),
+            size(out([1.;1.;;])[2]) == (1, 1),
         )
     end
 end