Define broadcasting behavior for KDEs

Author: andreasnoack

src/KernelDensity.jl

@@ -12,6 +12,9 @@ export kde, kde_lscv, UnivariateKDE, BivariateKDE, InterpKDE, pdf
 
 abstract type AbstractKDE end
 
+# define broadcasting behavior
+Base.Broadcast.broadcastable(x::AbstractKDE) = Ref(x)
+
 include("univariate.jl")
 include("bivariate.jl")
 include("interp.jl")

test/bivariate.jl

@@ -64,3 +64,8 @@ end
 k11 = kde([0.0 0.0; 1.0 1.0], (r,r), bandwidth=(1,1), weights=[0,1])
 k12 = kde([1.0 1.0], (r,r), bandwidth=(1,1))
 @test k11.density ≈ k12.density
+
+@testset "broadcasting. Issue 63" begin
+    xyvals = [0.0, 1.0]
+    @test pdf.(k11, xyvals, xyvals) == map(((x, y,)) -> pdf(k11, x, y), xyvals, xyvals)
+end

test/univariate.jl

@@ -69,3 +69,7 @@ abs_tol = eps(Float64)
 
 minimizer = @inferred KernelDensity.optimize(x -> (x - 1)^2, 0, 2)
 @test abs(minimizer - 1) <= abs_tol + rel_tol * abs(minimizer)
+
+@testset "broadcasting" begin
+    @test pdf(k11, [0.0, 1.0]) == map(t -> pdf(k11, t), [0.0, 1.0])
+end