Breadth FIrst Tree Traversal : Convert live test into Jest test.

lvlte · lvlte · commit e3d605c52827 · 2021-10-11T14:48:06.000+02:00
diff --git a/Trees/BreadthFirstTreeTraversal.js b/Trees/BreadthFirstTreeTraversal.js
@@ -22,7 +22,7 @@ class BinaryTree {
     for (let i = 1; i <= h; i++) {
       this.traverseLevel(this.root, i)
     }
-    return this.traversal.toLocaleString()
+    return this.traversal
   }
 
   // Computing the height of the tree
@@ -48,19 +48,4 @@ class BinaryTree {
   }
 }
 
-const binaryTree = new BinaryTree()
-const root = new Node(7)
-root.left = new Node(5)
-root.right = new Node(8)
-root.left.left = new Node(3)
-root.left.right = new Node(6)
-root.right.right = new Node(9)
-binaryTree.root = root
-
-console.log(binaryTree.breadthFirst())
-
-//            7
-//           / \
-//          5   8
-//         / \   \
-//        3   6   9
+export { BinaryTree, Node }
diff --git a/Trees/test/BreadthFirstTreeTraversal.test.js b/Trees/test/BreadthFirstTreeTraversal.test.js
@@ -0,0 +1,27 @@
+import { BinaryTree, Node } from '../BreadthFirstTreeTraversal'
+
+describe('Breadth First Tree Traversal', () => {
+  const binaryTree = new BinaryTree()
+
+  const root = new Node(7)
+  root.left = new Node(5)
+  root.right = new Node(8)
+  root.left.left = new Node(3)
+  root.left.right = new Node(6)
+  root.right.right = new Node(9)
+  binaryTree.root = root
+
+  // Vizualization :
+  //
+  //            7
+  //           / \
+  //          5   8
+  //         / \   \
+  //        3   6   9
+
+  it('Binary tree - Level order traversal', () => {
+    expect(binaryTree.traversal).toStrictEqual([])
+    const traversal = binaryTree.breadthFirst()
+    expect(traversal).toStrictEqual([7, 5, 8, 3, 6, 9])
+  })
+})
