Added a broadcastable method on 0.7 and tests for scalar broadcasting.

invenia · Jan 4, 2019 · 94eeb73 · 94eeb73
1 parent 810b8f4
commit 94eeb73
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/Intervals.jl b/src/Intervals.jl
@@ -16,6 +16,10 @@ abstract type AbstractInterval{T} end
 
 Base.eltype(::AbstractInterval{T}) where {T} = T
 
+if VERSION >= v"0.7"
+    Base.broadcastable(x::AbstractInterval) = Ref(x)
+end
+
 include("inclusivity.jl")
 include("endpoint.jl")
 include("interval.jl")

diff --git a/test/anchoredinterval.jl b/test/anchoredinterval.jl
@@ -590,4 +590,16 @@ using Intervals: canonicalize
             end
         end
     end
+
+    @testset "broadcasting" begin
+        # Check that we can broadcast a vector of anchored intervals against a scalar
+
+        a = [
+            HourEnding(DateTime(2016, 8, 11)),
+            HourEnding(DateTime(2016, 8, 12)),
+        ]
+
+        mask = a .== a[2]
+        @test mask == [false, true]
+    end
 end