|
| 1 | +package javatests.com.williamfiset.algorithms.datastructures.balancedtree; |
| 2 | + |
| 3 | +import static org.junit.Assert.*; |
| 4 | + |
| 5 | +import com.williamfiset.algorithms.datastructures.balancedtree.AVLTreeRecursive; |
| 6 | +import java.util.ArrayList; |
| 7 | +import java.util.Collections; |
| 8 | +import java.util.List; |
| 9 | +import java.util.TreeSet; |
| 10 | +import org.junit.Before; |
| 11 | +import org.junit.Test; |
| 12 | + |
| 13 | +public class AVLTreeTest { |
| 14 | + |
| 15 | + static final int MAX_RAND_NUM = +100000; |
| 16 | + static final int MIN_RAND_NUM = -100000; |
| 17 | + |
| 18 | + static final int TEST_SZ = 2500; |
| 19 | + |
| 20 | + private AVLTreeRecursive<Integer> tree; |
| 21 | + |
| 22 | + @Before |
| 23 | + public void setup() { |
| 24 | + tree = new AVLTreeRecursive<>(); |
| 25 | + } |
| 26 | + |
| 27 | + @Test |
| 28 | + public void testNullInsertion() { |
| 29 | + assertFalse(tree.insert(null)); |
| 30 | + } |
DEB9
code> | 31 | + |
| 32 | + @Test |
| 33 | + public void testNullRemoval() { |
| 34 | + assertFalse(tree.remove(null)); |
| 35 | + } |
| 36 | + |
| 37 | + @Test |
| 38 | + public void testTreeContainsNull() { |
| 39 | + assertFalse(tree.contains(null)); |
| 40 | + } |
| 41 | + |
| 42 | + @Test |
| 43 | + public void testLeftLeftCase() { |
| 44 | + |
| 45 | + tree.insert(3); |
| 46 | + tree.insert(2); |
| 47 | + tree.insert(1); |
| 48 | + |
| 49 | + assertEquals(2, tree.root.value.intValue()); |
| 50 | + assertEquals(1, tree.root.left.value.intValue()); |
| 51 | + assertEquals(3, tree.root.right.value.intValue()); |
| 52 | + |
| 53 | + assertNull(tree.root.left.left); |
| 54 | + assertNull(tree.root.left.right); |
| 55 | + assertNull(tree.root.right.left); |
| 56 | + assertNull(tree.root.right.right); |
| 57 | + } |
| 58 | + |
| 59 | + @Test |
| 60 | + public void testLeftRightCase() { |
| 61 | + |
| 62 | + tree.insert(3); |
| 63 | + tree.insert(1); |
| 64 | + tree.insert(2); |
| 65 | + |
| 66 | + assertEquals(2, tree.root.value.intValue()); |
| 67 | + assertEquals(1, tree.root.left.value.intValue()); |
| 68 | + assertEquals(3, tree.root.right.value.intValue()); |
| 69 | + |
| 70 | + assertNull(tree.root.left.left); |
| 71 | + assertNull(tree.root.left.right); |
| 72 | + assertNull(tree.root.right.left); |
| 73 | + assertNull(tree.root.right.right); |
| 74 | + } |
| 75 | + |
| 76 | + @Test |
| 77 | + public void testRightRightCase() { |
| 78 | + |
| 79 | + tree.insert(1); |
| 80 | + tree.insert(2); |
| 81 | + tree.insert(3); |
| 82 | + |
| 83 | + assertEquals(2, tree.root.value.intValue()); |
| 84 | + assertEquals(1, tree.root.left.value.intValue()); |
| 85 | + assertEquals(3, tree.root.right.value.intValue()); |
| 86 | + |
| 87 | + assertNull(tree.root.left.left); |
| 88 | + assertNull(tree.root.left.right); |
| 89 | + assertNull(tree.root.right.left); |
| 90 | + assertNull(tree.root.right.right); |
| 91 | + } |
| 92 | + |
| 93 | + @Test |
| 94 | + public void testRightLeftCase() { |
| 95 | + |
| 96 | + tree.insert(1); |
| 97 | + tree.insert(3); |
| 98 | + tree.insert(2); |
| 99 | + |
| 100 | + assertEquals(2, tree.root.value.intValue()); |
| 101 | + assertEquals(1, tree.root.left.value.intValue()); |
| 102 | + assertEquals(3, tree.root.right.value.intValue()); |
| 103 | + |
| 104 | + assertNull(tree.root.left.left); |
| 105 | + assertNull(tree.root.left.right); |
| 106 | + assertNull(tree.root.right.left); |
| 107 | + assertNull(tree.root.right.right); |
| 108 | + } |
| 109 | + |
| 110 | + @Test |
| 111 | + public void testRandomizedBalanceFactorTest() { |
| 112 | + for (int i = 0; i < TEST_SZ; i++) { |
| 113 | + tree.insert(randValue()); |
| 114 | + assertTrue(validateBalanceFactorValues(tree.root)); |
| 115 | + } |
| 116 | + } |
| 117 | + |
| 118 | + // Make sure all balance factor values are either -1, 0 or +1 |
| 119 | + static boolean validateBalanceFactorValues(AVLTreeRecursive.Node node) { |
| 120 | + if (node == null) return true; |
| 121 | + if (node.bf > +1 || node.bf < -1) return false; |
| 122 | + return validateBalanceFactorValues(node.left) && validateBalanceFactorValues(node.right); |
| 123 | + } |
| 124 | + |
| 125 | + @Test |
| 126 | + public void testRandomizedValueInsertionsAgainstTreeSet() { |
| 127 | + |
| 128 | + TreeSet<Integer> set = new TreeSet<>(); |
| 129 | + for (int i = 0; i < TEST_SZ; i++) { |
| 130 | + int v = randValue(); |
| 131 | + assertEquals(set.add(v), tree.insert(v)); |
| 132 | + assertEquals(set.size(), tree.size()); |
| 133 | + assertTrue(tree.validateBSTInvarient(tree.root)); |
| 134 | + } |
| 135 | + } |
| 136 | + |
| 137 | + @Test |
| 138 | + public void testTreeHeight() { |
| 139 | + for (int n = 1; n <= TEST_SZ; n++) { |
| 140 | + |
| 141 | + tree.insert(randValue()); |
| 142 | + int height = tree.height(); |
| 143 | + |
| 144 | + // Get an upper bound on what the maximum height of |
| 145 | + // an AVL tree should be. Values were taken from: |
| 146 | + // https://en.wikipedia.org/wiki/AVL_tree#Comparison_to_other_structures |
| 147 | + double c = 1.441; |
| 148 | + double b = -0.329; |
| 149 | + double upperBound = c * (Math.log(n + 2.0) / Math.log(2)) + b; |
| 150 | + |
| 151 | + assertTrue(height < upperBound); |
| 152 | + } |
| 153 | + } |
| 154 | + |
| 155 | + @Test |
| 156 | + public void randomRemoveTests() { |
| 157 | + TreeSet<Integer> ts = new TreeSet<>(); |
| 158 | + for (int i = 0; i < TEST_SZ; i++) { |
| 159 | + |
| 160 | + int size = i; |
| 161 | + List<Integer> lst = genRandList(size); |
| 162 | + for (Integer value : lst) { |
| 163 | + tree.insert(value); |
| 164 | + ts.add(value); |
| 165 | + } |
| 166 | + Collections.shuffle(lst); |
| 167 | + |
| 168 | + // Remove all the elements we just placed in the tree. |
| 169 | + for (int j = 0; j < size; j++) { |
| 170 | + |
| 171 | + Integer value = lst.get(j); |
| 172 | + |
| 173 | + assertEquals(ts.remove(value), tree.remove(value)); |
| 174 | + assertFalse(tree.contains(value)); |
| 175 | + assertEquals(size - j - 1, tree.size()); |
| 176 | + } |
| 177 | + |
| 178 | + assertTrue(tree.isEmpty()); |
| 179 | + } |
| 180 | + } |
| 181 | + |
| 182 | + static List<Integer> genRandList(int sz) { |
| 183 | + List<Integer> lst = new ArrayList<>(sz); |
| 184 | + for (int i = 0; i < sz; i++) lst.add(i); // unique values. |
| 185 | + Collections.shuffle(lst); |
| 186 | + return lst; |
| 187 | + } |
| 188 | + |
| 189 | + public static int randValue() { |
| 190 | + return (int) (Math.random() * MAX_RAND_NUM * 2) + MIN_RAND_NUM; |
| 191 | + } |
| 192 | +} |
0 commit comments