@Test
public void bstRemove() {
BST tree = new BST();
Student s = new Student("HeyySurname", "HeyyName", 2, 120, 10.1f, "HeyyStreet");
tree.insert(s.getId(), s);
assertEquals(1, tree.size());
Student s1 = new Student("HeyySurname1", "HeyyName1", 2, 99, 10.1f, "HeyyStreet");
tree.insert(s1.getId(), s1);
Student s2 = new Student("HeyySurname2", "HeyyName2", 2, 97, 10.1f, "HeyyStreet");
tree.insert(s2.getId(), s2);
Student s3 = new Student("HeyySurname3", "HeyyName3", 2, 98, 10.1f, "HeyyStreet");
tree.insert(s3.getId(), s3);
Student s4 = new Student("HeyySurname4", "HeyyName4", 2, 131, 10.1f, "HeyyStreet");
tree.insert(s4.getId(), s4);
Student returnedStudent = tree.remove(s.getId()); // Two children
assertEquals(returnedStudent, s);
java.io.ByteArrayOutputStream out = new java.io.ByteArrayOutputStream();
System.setOut(new java.io.PrintStream(out));
tree.traverse();
String outputStringTraversal = out.toString().split("\n")[2];
assertEquals("97 98 99 131 ", outputStringTraversal);
java.io.ByteArrayOutputStream out1 = new java.io.ByteArrayOutputStream();
System.setOut(new java.io.PrintStream(out1));
tree.remove(s2.getId()); // one child
tree.traverse();
outputStringTraversal = out1.toString().split("\n")[2];
assertEquals("98 99 131 ", outputStringTraversal);
java.io.ByteArrayOutputStream out2 = new java.io.ByteArrayOutputStream();
System.setOut(new java.io.PrintStream(out2));
tree.remove(s4.getId()); // no children
tree.traverse();
System.setOut(new PrintStream(new FileOutputStream(FileDescriptor.out)));
outputStringTraversal = out2.toString().split("\n")[2];
assertEquals("98 99 ", outputStringTraversal);
}
