From cba4f72913f42dcac471669dd8553e2e16a3177c Mon Sep 17 00:00:00 2001
From: Philip Offtermatt <57488781+p-offtermatt@users.noreply.github.com>
Date: Mon, 7 Aug 2023 16:12:25 +0200
Subject: [PATCH] Add mapRemoveAll to basic spells

---
 examples/spells/basicSpells.qnt | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/examples/spells/basicSpells.qnt b/examples/spells/basicSpells.qnt
index 32747e0f6..6bef90874 100644
--- a/examples/spells/basicSpells.qnt
+++ b/examples/spells/basicSpells.qnt
@@ -123,4 +123,21 @@ module basicSpells {
     assert(Map(3 -> 4, 7 -> 8) == Map(3 -> 4, 5 -> 6, 7 -> 8).mapRemove(5)),
     assert(Map() == Map().mapRemove(3)),
   }
-}
\ No newline at end of file
+
+  /// Removes a set of map entry.
+  ///
+  /// - @param __map a map to remove an entry from
+  /// - @param __keys a set of keys to remove from the map
+  /// - @returns a new map that contains all entries of __map
+  ///          that do not have a key in __keys
+  pure def mapRemoveAll(__map: a -> b, __keys: Set[a]): a -> b = {
+      __map.keys().exclude(__keys).mapBy(__k => __map.get(__k))
+  }
+
+  run mapRemoveAllTest =
+      val m = Map(3 -> 4, 5 -> 6, 7 -> 8)
+      all {
+          assert(m.mapRemoveAll(Set(5, 7)) == Map(3 -> 4)),
+          assert(m.mapRemoveAll(Set(5, 99999)) == Map(3 -> 4, 7 -> 8)),
+      }
+}