Equality for Non-ordered anchored intervals

invenia · Aug 14, 2018 · 400d570 · 400d570
1 parent 9f0fa69
commit 400d570
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/src/anchoredinterval.jl b/src/anchoredinterval.jl
@@ -215,6 +215,10 @@ end
 
 ##### EQUALITY #####
 
+function Base.:(==)(a::AnchoredInterval{P, T}, b::AnchoredInterval{P, T}) where {P, T}
+    return anchor(a) == anchor(b) && inclusivity(a) == inclusivity(b)
+end
+
 # Required for min/max of AnchoredInterval{LaxZonedDateTime} when the anchor is AMB or DNE
 function Base.isless(a::AnchoredInterval{P, T}, b::AnchoredInterval{P, T}) where {P, T}
     return (

diff --git a/test/anchoredinterval.jl b/test/anchoredinterval.jl
@@ -37,13 +37,17 @@ using Intervals: canonicalize
     end
 
     @testset "non-ordered" begin
-        interval = AnchoredInterval{-Inf}(Inf)
-        @test isequal(first(interval), NaN)
-        @test isequal(last(interval), Inf)
-
-        interval = AnchoredInterval{Inf}(-Inf)
-        @test isequal(first(interval), -Inf)
-        @test isequal(last(interval), NaN)
+        ending = AnchoredInterval{-Inf}(Inf)
+        @test isequal(first(ending), NaN)
+        @test isequal(last(ending), Inf)
+        @test ending == ending
+        @test isequal(ending, ending)
+
+        beginning = AnchoredInterval{Inf}(-Inf)
+        @test isequal(first(beginning), -Inf)
+        @test isequal(last(beginning), NaN)
+        @test beginning == beginning
+        @test isequal(beginning, beginning)
     end
 
     @testset "hash" begin