55 lines
1.4 KiB
Gleam
55 lines
1.4 KiB
Gleam
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")])
|
|
}
|