test Diagonal covariance

Author: mohamed82008

test/distributions.jl

@@ -176,26 +176,34 @@ separator()
         # Vector case
         DistSpec(:MvNormal, (mean, cov_mat), norm_val_vec),
         DistSpec(:MvNormal, (mean, cov_vec), norm_val_vec),
+        DistSpec(:MvNormal, (mean, Diagonal(cov_vec)), norm_val_vec),
         DistSpec(:MvNormal, (mean, cov_num), norm_val_vec),
         DistSpec(:((m, v) -> MvNormal(m, v*I)), (mean, cov_num), norm_val_vec),
         DistSpec(:MvNormal, (cov_mat,), norm_val_vec),
         DistSpec(:MvNormal, (cov_vec,), norm_val_vec),
+        DistSpec(:MvNormal, (Diagonal(cov_vec),), norm_val_vec),
         DistSpec(:(cov_num -> MvNormal(dim, cov_num)), (cov_num,), norm_val_vec),
         DistSpec(:MvLogNormal, (mean, cov_mat), norm_val_vec),
         DistSpec(:MvLogNormal, (mean, cov_vec), norm_val_vec),
+        DistSpec(:MvLogNormal, (mean, Diagonal(cov_vec)), norm_val_vec),
         DistSpec(:MvLogNormal, (mean, cov_num), norm_val_vec),
         DistSpec(:MvLogNormal, (cov_mat,), norm_val_vec),
         DistSpec(:MvLogNormal, (cov_vec,), norm_val_vec),
+        DistSpec(:MvLogNormal, (Diagonal(cov_vec),), norm_val_vec),
         DistSpec(:(cov_num -> MvLogNormal(dim, cov_num)), (cov_num,), norm_val_vec),
         # Matrix case
         DistSpec(:MvNormal, (mean, cov_vec), norm_val_mat),
+        DistSpec(:MvNormal, (mean, Diagonal(cov_vec)), norm_val_mat),
         DistSpec(:MvNormal, (mean, cov_num), norm_val_mat),
         DistSpec(:((m, v) -> MvNormal(m, v*I)), (mean, cov_num), norm_val_mat),
         DistSpec(:MvNormal, (cov_vec,), norm_val_mat),
+        DistSpec(:MvNormal, (Diagonal(cov_vec),), norm_val_mat),
         DistSpec(:(cov_num -> MvNormal(dim, cov_num)), (cov_num,), norm_val_mat),
         DistSpec(:MvLogNormal, (mean, cov_vec), norm_val_mat),
+        DistSpec(:MvLogNormal, (mean, Diagonal(cov_vec)), norm_val_mat),
         DistSpec(:MvLogNormal, (mean, cov_num), norm_val_mat),
         DistSpec(:MvLogNormal, (cov_vec,), norm_val_mat),
+        DistSpec(:MvLogNormal, (Diagonal(cov_vec),), norm_val_mat),
         DistSpec(:(cov_num -> MvLogNormal(dim, cov_num)), (cov_num,), norm_val_mat),
     ]
 

test/test_utils.jl

@@ -10,6 +10,7 @@ struct DistSpec{Tθ<:Tuple, Tx}
 end
 
 vectorize(v::Number) = [v]
+vectorize(v::Diagonal) = v.diag
 vectorize(v) = vec(v)
 pack(vals...) = reduce(vcat, vectorize.(vals))
 @generated function unpack(x, vals...)
@@ -22,6 +23,9 @@ pack(vals...) = reduce(vcat, vectorize.(vals))
         elseif T <: Vector
             push!(unpacked, :(x[$ind:$ind+length(vals[$i])-1]))
             ind = :($ind + length(vals[$i]))
+        elseif T <: Diagonal
+            push!(unpacked, :(Diagonal(x[$ind:$ind+size(vals[$i],1)-1])))
+            ind = :($ind + size(vals[$i],1))
         elseif T <: Matrix
             push!(unpacked, :(reshape(x[$ind:($ind+length(vals[$i])-1)], size(vals[$i]))))
             ind = :($ind + length(vals[$i]))