Product of Diracs (#223)

* Product of Diracs

* bump version

Author: cscherrer

Project.toml

@@ -1,7 +1,7 @@
 name = "MeasureTheory"
 uuid = "eadaa1a4-d27c-401d-8699-e962e1bbc33b"
 authors = ["Chad Scherrer <[email protected]> and contributors"]
-version = "0.17.0"
+version = "0.17.1"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

src/combinators/product.jl

@@ -2,6 +2,10 @@ function as(d::PowerMeasure)
     as(Array, as(d.parent), length.(d.axes)...)
 end
 
+function as(d::ProductMeasure{<:AbstractArray{<:Dirac}})
+    return asConst(testvalue.(marginals(d)))
+end
+
 function as(d::ProductMeasure{A}) where {A<:AbstractArray}
     mar = marginals(d)
     ts = map(as, mar)

test/runtests.jl

@@ -275,6 +275,12 @@ end
     end
 end
 
+@testset "Product of Diracs" begin
+    x = randn(3)
+    t = as(productmeasure(Dirac.(x))) 
+    @test transform(t, []) == x
+end
+
 # @testset "Univariate chain" begin
 #     ξ0 = 1.
 #     x = 1.2