Product of Diracs (#223)

* Product of Diracs * bump version
JuliaMath · Aug 14, 2022 · e4aa5cf · e4aa5cf · cscherrer · Aug 14, 2022
1 parent b42228f
commit e4aa5cf
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/Project.toml b/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"

diff --git a/src/combinators/product.jl b/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)

diff --git a/test/runtests.jl b/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