feat: add size() function

Adds a size() function to measure the tree size.
IMA-WorldHealth · Dec 21, 2020 · b5ec97d · b5ec97d
1 parent 81a063e
commit b5ec97d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -73,6 +73,12 @@ class Tree {
     return pruned;
   }
 
+  size() {
+    let size = 0;
+    this.walk(() => size++);
+    return size;
+  }
+
   toArray() {
     const array = [];
     this.walk((node) => array.push(node));

diff --git a/test/tree.js b/test/tree.js
@@ -117,6 +117,10 @@ test('#prune() should remove nodes based on pruning function', t => {
   t.false(afterIds.includes(5));
 });
 
+test('#size() should return the correct size of the tree', t => {
+  t.is(tree.size(), 7);
+});
+
 test('#walk() should be able to compute the balances of nodes', t => {
   const node1 = tree.find(1);
   const node4 = tree.find(4);