erlang-extraction

erlang-extraction/test/erlang_extraction_test.gleam

@@ -0,0 +1,54 @@
+import erlang_extraction.{type GbTree}
+import exercism/should
+import exercism/test_runner
+
+pub fn main() {
+  test_runner.main()
+}
+
+@external(erlang, "gb_trees", "to_list")
+fn to_list(tree: GbTree(k, v)) -> List(#(k, v))
+
+pub fn new_gb_tree_test() {
+  erlang_extraction.new_gb_tree()
+  |> to_list
+  |> should.equal([])
+}
+
+pub fn insert_int_string_test() {
+  erlang_extraction.new_gb_tree()
+  |> erlang_extraction.insert(1, "one")
+  |> erlang_extraction.insert(2, "two")
+  |> erlang_extraction.insert(3, "three")
+  |> to_list
+  |> should.equal([#(1, "one"), #(2, "two"), #(3, "three")])
+}
+
+pub fn insert_string_int_test() {
+  erlang_extraction.new_gb_tree()
+  |> erlang_extraction.insert("one", 1)
+  |> erlang_extraction.insert("two", 2)
+  |> erlang_extraction.insert("three", 3)
+  |> to_list
+  |> should.equal([#("one", 1), #("three", 3), #("two", 2)])
+}
+
+pub fn delete_test() {
+  erlang_extraction.new_gb_tree()
+  |> erlang_extraction.insert(1, "one")
+  |> erlang_extraction.insert(2, "two")
+  |> erlang_extraction.insert(3, "three")
+  |> erlang_extraction.delete(2)
+  |> to_list
+  |> should.equal([#(1, "one"), #(3, "three")])
+}
+
+pub fn delete_non_existing_test() {
+  erlang_extraction.new_gb_tree()
+  |> erlang_extraction.insert(1, "one")
+  |> erlang_extraction.insert(2, "two")
+  |> erlang_extraction.insert(3, "three")
+  |> erlang_extraction.delete(4)
+  |> to_list
+  |> should.equal([#(1, "one"), #(2, "two"), #(3, "three")])
+}