Add a few tests.

1 files changed, 8 insertions, 3 deletions

diff --git a/src/ts.erl b/src/ts.erl
index a4d8107..841fdd7 100644
--- a/src/ts.erl
+++ b/src/ts.erl
@@ -4,8 +4,8 @@
 -export_type([tree_store/0]).
 -export([new/0, store/3, append/3, truncate/2, retrieve/2, retrieve_hash/2, delete/2]).
 
-%% FIXME: keep the arrays in an array? or maybe an array of binaries?
-%% hashes do have fixed lenght.
+%% FIXME: Keep the arrays in an array instead of in a map? Or maybe an
+%% array of binaries? Hashes do have fixed lenght.
 -record(tree_store, {arrays :: map()}). % Map of arrays, indexed by layer.
 -type tree_store() :: #tree_store{}.
 
@@ -69,4 +69,9 @@ store_test_() ->
     S0 = store(S, {0,0}, <<00>>),
     S1 = store(S0, {0,1}, <<01>>),
     S2 = store(S1, {1,0}, <<10>>),
-    [?_assertEqual(retrieve(S2, {1,0}), element(3, T))].
+    [?_assertEqual(retrieve(S2, {1,0}), element(3, T)),
+     ?_assertEqual(retrieve(S2, {0,1}), element(2, T)),
+     ?_assertEqual(retrieve(S2, {0,0}), element(1, T)),
+     ?_assertEqual(retrieve_hash(S2, {3,0}), undefined),
+     ?_assertEqual(retrieve_hash(S1, {1,0}), undefined)
+    ].