Merge pull request #35 from devmotion/params

Readd accidentally removed methods

Author: mohamed82008

src/multivariate.jl

@@ -116,6 +116,7 @@ struct TuringDiagMvNormal{Tm<:AbstractVector, Tσ<:AbstractVector} <: Continuous
     σ::Tσ
 end
 
+Distributions.params(d::TuringDiagMvNormal) = (d.m, d.σ)
 Base.length(d::TuringDiagMvNormal) = length(d.m)
 Base.size(d::TuringDiagMvNormal) = (length(d),)
 Distributions.rand(d::TuringDiagMvNormal, n::Int...) = rand(Random.GLOBAL_RNG, d, n...)
@@ -128,6 +129,7 @@ struct TuringScalMvNormal{Tm<:AbstractVector, Tσ<:Real} <: ContinuousMultivaria
     σ::Tσ
 end
 
+Distributions.params(d::TuringScalMvNormal) = (d.m, d.σ)
 Base.length(d::TuringScalMvNormal) = length(d.m)
 Base.size(d::TuringScalMvNormal) = (length(d),)
 Distributions.rand(d::TuringScalMvNormal, n::Int...) = rand(Random.GLOBAL_RNG, d, n...)

test/others.jl

@@ -125,3 +125,22 @@ end
     B = copy(A)
     @test DistributionsAD.zygote_ldiv(A, B) == A \ B
 end
+
+@testset "Entropy" begin
+    sigmas = exp.(randn(10))
+    d1 = TuringDiagMvNormal(zeros(10), sigmas)
+    d2 = MvNormal(zeros(10), sigmas)
+
+    @test entropy(d1) == entropy(d2)
+end
+
+@testset "Params" begin
+    m = rand(10)
+    sigmas = randexp(10)
+    
+    d = TuringDiagMvNormal(m, sigmas)
+    @test params(d) == (m, sigmas)
+
+    d = TuringScalMvNormal(m, sigmas[1])
+    @test params(d) == (m, sigmas[1])
+end