Add get call and associated test

1 files changed, 13 insertions, 0 deletions

diff --git a/test/mst_test.exs b/test/mst_test.exs
index 68886c7..6615b6e 100644
--- a/test/mst_test.exs
+++ b/test/mst_test.exs
@@ -11,6 +11,14 @@ defmodule ShardTest.MST do
     z = Enum.reduce(Enum.shuffle(0..1000), MST.new(),
                     fn i, acc -> MST.insert(acc, i, i) end)
 
+    for i <- 0..1000 do
+      assert MST.get(y, i) == i
+      assert MST.get(z, i) == i
+    end
+    assert MST.get(y, 9999) == nil
+    assert MST.get(z, -1001) == nil
+    assert MST.get(z, 1.01) == nil
+
     IO.puts "y.root: #{y.root|>Base.encode16}"
     IO.puts "z.root: #{z.root|>Base.encode16}"
     assert y.root == z.root
@@ -28,6 +36,11 @@ defmodule ShardTest.MST do
     z = Enum.reduce(Enum.shuffle(items), MST.new(),
                     fn {k, v}, acc -> MST.insert(acc, k, v) end)
 
+    for {k, v} <- items do
+      assert MST.get(y, k) == v
+      assert MST.get(z, k) == v
+    end
+
     IO.puts "y.root: #{y.root|>Base.encode16}"
     IO.puts "z.root: #{z.root|>Base.encode16}"
     assert y.root == z.root