@@ -14,6 +14,8 @@ public void testAddElement() {
14
14
heap .addElement (8 );
15
15
16
16
assertEquals (8 , heap .removeElement ()); // Max element should be 8
17
+ assertEquals (5 , heap .removeElement ());
18
+ assertEquals (3 , heap .removeElement ());
17
19
}
18
20
19
21
@ Test
@@ -38,6 +40,10 @@ public void testAddElementStrings() {
38
40
heap .addElement ("y" );
39
41
40
42
assertEquals ("z" , heap .removeElement ()); // Max element should be z
43
+ assertEquals ("y" , heap .removeElement ());
44
+ assertEquals ("x" , heap .removeElement ());
45
+ assertEquals ("b" , heap .removeElement ());
46
+ assertEquals ("a" , heap .removeElement ());
41
47
}
42
48
43
49
@ Test
@@ -91,10 +97,25 @@ public void testForCompareChildAndSwap() {
91
97
heap .addElement (element );
92
98
}
93
99
94
- // Assert the top 4 elemets are extracted correctly
95
100
assertEquals (95 , heap .removeElement ());
96
101
assertEquals (94 , heap .removeElement ());
97
102
assertEquals (88 , heap .removeElement ());
98
103
assertEquals (84 , heap .removeElement ());
104
+ assertEquals (62 , heap .removeElement ());
105
+ assertEquals (61 , heap .removeElement ());
106
+ assertEquals (59 , heap .removeElement ());
107
+ assertEquals (52 , heap .removeElement ());
108
+ assertEquals (48 , heap .removeElement ());
109
+ assertEquals (40 , heap .removeElement ());
110
+ assertEquals (37 , heap .removeElement ());
111
+ assertEquals (33 , heap .removeElement ());
112
+ assertEquals (33 , heap .removeElement ());
113
+ assertEquals (29 , heap .removeElement ());
114
+ assertEquals (28 , heap .removeElement ());
115
+ assertEquals (15 , heap .removeElement ());
116
+ assertEquals (13 , heap .removeElement ());
117
+ assertEquals (12 , heap .removeElement ());
118
+ assertEquals (5 , heap .removeElement ());
119
+ assertEquals (2 , heap .removeElement ());
99
120
}
100
121
}
0 commit comments