free UnivariateFinite from having a supertype

Author: ablaom

src/methods.jl

@@ -371,3 +371,9 @@ function Dist.fit(d::Type{<:UnivariateFinite},
 end
 
 
+# # BROADCASTING OVER SINGLE UNIVARIATE FINITE
+
+# This mirrors behaviour assigned Distributions.Distribution objects,
+# which allows `pdf.(d::UnivariateFinite, support(d))` to work. 
+
+Broadcast.broadcastable(d::UnivariateFinite) = Ref(d)

test/methods.jl

@@ -121,6 +121,11 @@ A, S, Q, F = V[1], V[2], V[3], V[4]
     @test pdf(d, "class_1") == 0.7
 end
 
+@testset "broadcasting pdf over single UnivariateFinite object" begin
+    d = UnivariateFinite(["a", "b"], [0.1, 0.9], pool=missing);
+    @test pdf.(d, ["a", "b"]) == [0.1, 0.9]
+end 
+
 @testset "constructor arguments not categorical values" begin
     @test_throws ArgumentError UnivariateFinite(Dict('f'=>0.7, 'q'=>0.2))
     @test_throws ArgumentError UnivariateFinite(Dict('f'=>0.7, 'q'=>0.2),
@@ -277,7 +282,6 @@ end
     # @test v ≈ v_close
 end
 
-
 end # module
 
 true