Refactored comparisons and sets

2 files changed, 45 insertions, 0 deletions

diff --git a/std/Set/NumberSet.bruijn b/std/Set/NumberSet.bruijn
new file mode 100644
index 0000000..220e2dc
--- /dev/null
+++ b/std/Set/NumberSet.bruijn
@@ -0,0 +1,22 @@
+# MIT License, Copyright (c) 2024 Marvin Borner
+
+:input std/Set
+
+:import std/Number T
+
+# adds a number of a set
+add T.<?>add ⧗ Number → NumberSet → NumberSet
+
+# returns true if a number is in a set
+has? T.<?>has? ⧗ Number → NumberSet → Boolean
+
+:test (has? (+5) (add (+5) empty)) ([[1]])
+:test (has? (+5) empty) ([[0]])
+
+# converts a list to a set
+list→set T.<?>list→set ⧗ (List Number) → NumberSet
+
+:test (has? (+0) (list→set ((+5) : ((+3) : ((+2) : ((+1) : {}(+0))))))) ([[1]])
+:test (has? (+5) (list→set ((+5) : ((+3) : ((+2) : ((+1) : {}(+0))))))) ([[1]])
+:test (has? (+6) (list→set ((+5) : ((+3) : ((+2) : ((+1) : {}(+0))))))) ([[0]])
+:test (has? (+7) (list→set ((+5) : ((+7) : {}(+1))))) ([[1]])
diff --git a/std/Set/StringSet.bruijn b/std/Set/StringSet.bruijn
new file mode 100644
index 0000000..4bee345
--- /dev/null
+++ b/std/Set/StringSet.bruijn
@@ -0,0 +1,23 @@
+# MIT License, Copyright (c) 2024 Marvin Borner
+# TODO: hash instead of comparing
+
+:input std/Set
+
+:import std/String S
+
+# adds a number of a set
+add S.<?>add ⧗ String → StringSet → StringSet
+
+# returns true if a number is in a set
+has? S.<?>has? ⧗ String → StringSet → Boolean
+
+:test (has? "abc" (add "abc" empty)) ([[1]])
+:test (has? "abc" empty) ([[0]])
+
+# converts a list to a set
+list→set S.<?>list→set ⧗ (List String) → StringSet
+
+:test (has? "0" (list→set ("a" : ("b" : ("d" : ("c" : {}"0")))))) ([[1]])
+:test (has? "a" (list→set ("a" : ("b" : ("d" : ("c" : {}"0")))))) ([[1]])
+:test (has? "e" (list→set ("a" : ("b" : ("d" : ("c" : {}"0")))))) ([[0]])
+:test (has? "c" (list→set ("a" : ("c" : {}"b")))) ([[1]])