test: refactor tests

ridge-kimani · ridge-kimani · commit dd02c5c56446 · 2025-01-07T21:06:15.000+03:00
diff --git a/Data-Structures/Tree/test/BinarySearchTree.test.js b/Data-Structures/Tree/test/BinarySearchTree.test.js
@@ -5,85 +5,61 @@ describe('Binary Search Tree', () => {
 
   beforeEach(() => {
     tree = new Tree()
-  })
-
-  test('should add values to the tree', () => {
     tree.addValue(10)
     tree.addValue(5)
     tree.addValue(15)
+    tree.addValue(3)
+    tree.addValue(8)
+  })
 
-    expect(tree.search(10)).toBe(10)
+  test('should add values to the tree', () => {
+    tree.addValue(12)
+
+    expect(tree.search(12)).toBe(12)
     expect(tree.search(5)).toBe(5)
     expect(tree.search(15)).toBe(15)
   })
 
   test('should perform in-order traversal', () => {
     const values = []
     const output = (val) => values.push(val)
-
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
-    tree.addValue(3)
-    tree.addValue(8)
-
     tree.traverse(output)
     expect(values).toEqual([3, 5, 8, 10, 15])
   })
 
   test('should remove leaf nodes correctly', () => {
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
-
     tree.removeValue(5)
     expect(tree.search(5)).toBeNull()
   })
 
   test('should remove nodes with one child correctly', () => {
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
     tree.addValue(12)
-
     tree.removeValue(15)
+
     expect(tree.search(15)).toBeNull()
     expect(tree.search(12)).toBe(12)
   })
 
   test('should remove nodes with two children correctly', () => {
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
-    tree.addValue(12)
     tree.addValue(18)
-
     tree.removeValue(15)
+
     expect(tree.search(15)).toBeNull()
-    expect(tree.search(12)).toBe(12)
     expect(tree.search(18)).toBe(18)
   })
 
   test('should return null for non-existent values', () => {
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
-
     expect(tree.search(20)).toBeNull()
     expect(tree.search(0)).toBeNull()
   })
 
   test('should handle removal of root node correctly', () => {
-    tree.addValue(10)
-    tree.addValue(5)
-    tree.addValue(15)
-
     tree.removeValue(10)
     expect(tree.search(10)).toBeNull()
   })
 
   test('should handle empty tree gracefully', () => {
-    expect(tree.search(10)).toBeNull()
-    tree.removeValue(10) // Should not throw
+    expect(tree.search(22)).toBeNull()
+    tree.removeValue(22) // Should not throw
   })
 })
