diff --git a/core-java/src/main/java/com/baeldung/tree/BinaryTree.java b/core-java/src/main/java/com/baeldung/tree/BinaryTree.java index 3c5f5f4e10..3cc496e348 100644 --- a/core-java/src/main/java/com/baeldung/tree/BinaryTree.java +++ b/core-java/src/main/java/com/baeldung/tree/BinaryTree.java @@ -39,7 +39,6 @@ public class BinaryTree { parent = current; } - } public boolean isEmpty() { diff --git a/core-java/src/test/java/com/baeldung/tree/BinaryTreeTest.java b/core-java/src/test/java/com/baeldung/tree/BinaryTreeTest.java index 0b123a9684..2c20c730df 100644 --- a/core-java/src/test/java/com/baeldung/tree/BinaryTreeTest.java +++ b/core-java/src/test/java/com/baeldung/tree/BinaryTreeTest.java @@ -10,15 +10,7 @@ public class BinaryTreeTest { @Test public void givenABinaryTree_WhenAddingElements_ThenTreeNotEmpty() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); assertTrue(!bt.isEmpty()); } @@ -26,66 +18,28 @@ public class BinaryTreeTest { @Test public void givenABinaryTree_WhenAddingElements_ThenTreeContainsThoseElements() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); assertTrue(bt.containsNode(6)); assertTrue(bt.containsNode(4)); - assertTrue(bt.containsNode(8)); - assertTrue(bt.containsNode(3)); - assertTrue(bt.containsNode(5)); - assertTrue(bt.containsNode(7)); - assertTrue(bt.containsNode(9)); assertFalse(bt.containsNode(1)); - assertFalse(bt.containsNode(10)); } @Test public void givenABinaryTree_WhenDeletingElements_ThenTreeDoesNotContainThoseElements() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); assertTrue(bt.containsNode(9)); bt.delete(9); assertFalse(bt.containsNode(9)); - - assertTrue(bt.containsNode(6)); - bt.delete(6); - assertFalse(bt.containsNode(6)); - - assertTrue(bt.containsNode(4)); - bt.delete(4); - assertFalse(bt.containsNode(4)); } @Test public void givenABinaryTree_WhenTraversingInOrder_ThenPrintValues() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); bt.traverseInOrder(bt.root); } @@ -93,15 +47,7 @@ public class BinaryTreeTest { @Test public void givenABinaryTree_WhenTraversingPreOrder_ThenPrintValues() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); bt.traversePreOrder(bt.root); } @@ -109,15 +55,7 @@ public class BinaryTreeTest { @Test public void givenABinaryTree_WhenTraversingPostOrder_ThenPrintValues() { - BinaryTree bt = new BinaryTree(); - - bt.add(6); - bt.add(4); - bt.add(8); - bt.add(3); - bt.add(5); - bt.add(7); - bt.add(9); + BinaryTree bt = createBinaryTree(); bt.traversePostOrder(bt.root); } @@ -125,6 +63,12 @@ public class BinaryTreeTest { @Test public void givenABinaryTree_WhenTraversingLevelOrder_ThenPrintValues() { + BinaryTree bt = createBinaryTree(); + + bt.traverseLevelOrder(); + } + + private BinaryTree createBinaryTree() { BinaryTree bt = new BinaryTree(); bt.add(6); @@ -135,7 +79,7 @@ public class BinaryTreeTest { bt.add(7); bt.add(9); - bt.traverseLevelOrder(); + return bt; } }