add subtraction method on AtLeast2[Set, X] which yields NonEmptySet[X]

gerlichlab · Nov 16, 2024 · 6d0763c · 6d0763c
1 parent 112d1f9
commit 6d0763c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modules/pan/src/main/scala/collections.scala b/modules/pan/src/main/scala/collections.scala
@@ -101,6 +101,9 @@ object collections:
         infix def +(x: X): AtLeast2Set[X] =
           (xs + x).refineUnsafe[MinLength[2]]
 
+        def subtract(x: X)(using ord: Order[X]): NonEmptySet[X] =
+          NonEmptySet.fromSetUnsafe(SortedSet.from(xs - x)(using ord.toOrdering))
+
         /** With knowledge that the given container type is an set, we can use the underlying
           * collection's {@code .contains} member.
           *