Test for inclusion.

robrix · Jul 4, 2020 · 1aa0cff · 1aa0cff
1 parent 3e113a3
commit 1aa0cff
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/Data/IntervalSet.hs b/src/Data/IntervalSet.hs
@@ -32,9 +32,10 @@ insert :: Ord a => Interval I a -> IntervalSet a -> IntervalSet a
 insert i = \case
   Empty -> singleton i
   Branch b l i' g
-    | sup i < inf i' -> Branch b' (insert i l) i' g
-    | inf i < sup i' -> Branch b' l i' (insert i g)
-    | otherwise      -> merge b' l (i `union` i') g
+    | i `isSubintervalOf` i' -> Branch b l i' g
+    | sup i < inf i'         -> Branch b' (insert i l) i' g
+    | inf i < sup i'         -> Branch b' l i' (insert i g)
+    | otherwise              -> merge b' l (i `union` i') g
     where
     b' = i `union` b