Просмотр исходного кода

binary-search-tree: use generics

Frida Tveit 9 лет назад
Родитель
Сommit
74bf896f96
1 измененных файлов: 36 добавлений и 30 удалений
  1. 36
    30
      exercises/binary-search-tree/src/test/java/BinarySearchTreeTest.java

+ 36
- 30
exercises/binary-search-tree/src/test/java/BinarySearchTreeTest.java Просмотреть файл

@@ -9,39 +9,38 @@ import org.junit.Test;
9 9
 import org.junit.Before;
10 10
 
11 11
 public class BinarySearchTreeTest {
12
-    private BinarySearchTree<Integer> binarySearchTree;
13
-
14
-    @Before
15
-    public void setUp() {
16
-        binarySearchTree = new BinarySearchTree<>();
17
-    }
18 12
 
19 13
     @Test
20 14
     public void dataIsRetained() {
21
-        final int actual = 4;
22
-        binarySearchTree.insert(actual);
15
+        BinarySearchTree<Integer> binarySearchTree = new BinarySearchTree<>();
16
+
17
+        final int expected = 4;
18
+        binarySearchTree.insert(expected);
19
+
23 20
         final BinarySearchTree.Node<Integer> root = binarySearchTree.getRoot();
24 21
         assertNotNull(root);
25
-        final int expected = root.getData();
22
+
23
+        final int actual = root.getData();
26 24
         assertEquals(expected, actual);
27 25
     }
28 26
 
29 27
     @Ignore("Remove to run test")
30 28
     @Test
31 29
     public void insertsLess() {
32
-        final int expectedRoot = 4;
33
-        final int expectedLeft = 2;
30
+        BinarySearchTree<Character> binarySearchTree = new BinarySearchTree<>();
31
+        final char expectedRoot = '4';
32
+        final char expectedLeft = '2';
34 33
 
35 34
         binarySearchTree.insert(expectedRoot);
36 35
         binarySearchTree.insert(expectedLeft);
37 36
 
38
-        final BinarySearchTree.Node<Integer> root = binarySearchTree.getRoot();
37
+        final BinarySearchTree.Node<Character> root = binarySearchTree.getRoot();
39 38
         assertNotNull(root);
40
-        final BinarySearchTree.Node<Integer> left = root.getLeft();
39
+        final BinarySearchTree.Node<Character> left = root.getLeft();
41 40
         assertNotNull(left);
42 41
 
43
-        final int actualRoot = root.getData();
44
-        final int actualLeft = left.getData();
42
+        final char actualRoot = root.getData();
43
+        final char actualLeft = left.getData();
45 44
         assertEquals(expectedLeft, actualLeft);
46 45
         assertEquals(expectedRoot, actualRoot);
47 46
     }
@@ -49,19 +48,20 @@ public class BinarySearchTreeTest {
49 48
     @Ignore("Remove to run test")
50 49
     @Test
51 50
     public void insertsSame() {
52
-        final int expectedRoot = 4;
53
-        final int expectedLeft = 4;
51
+        BinarySearchTree<String> binarySearchTree = new BinarySearchTree<>();
52
+        final String expectedRoot = "4";
53
+        final String expectedLeft = "4";
54 54
 
55 55
         binarySearchTree.insert(expectedRoot);
56 56
         binarySearchTree.insert(expectedLeft);
57 57
 
58
-        final BinarySearchTree.Node<Integer> root = binarySearchTree.getRoot();
58
+        final BinarySearchTree.Node<String> root = binarySearchTree.getRoot();
59 59
         assertNotNull(root);
60
-        final BinarySearchTree.Node<Integer> left = root.getLeft();
60
+        final BinarySearchTree.Node<String> left = root.getLeft();
61 61
         assertNotNull(left);
62 62
 
63
-        final int actualRoot = root.getData();
64
-        final int actualLeft = left.getData();
63
+        final String actualRoot = root.getData();
64
+        final String actualLeft = left.getData();
65 65
         assertEquals(expectedLeft, actualLeft);
66 66
         assertEquals(expectedRoot, actualRoot);
67 67
     }
@@ -69,6 +69,7 @@ public class BinarySearchTreeTest {
69 69
     @Ignore("Remove to run test")
70 70
     @Test
71 71
     public void insertsRight() {
72
+        BinarySearchTree<Integer> binarySearchTree = new BinarySearchTree<>();
72 73
         final int expectedRoot = 4;
73 74
         final int expectedRight = 5;
74 75
 
@@ -89,6 +90,7 @@ public class BinarySearchTreeTest {
89 90
     @Ignore("Remove to run test")
90 91
     @Test
91 92
     public void createsComplexTree() {
93
+        BinarySearchTree<Integer> binarySearchTree = new BinarySearchTree<>();
92 94
         List<Integer> expected = Collections.unmodifiableList(
93 95
                 Arrays.asList(4, 2, 6, 1, 3, 5, 7)
94 96
         );
@@ -105,33 +107,36 @@ public class BinarySearchTreeTest {
105 107
     @Ignore("Remove to run test")
106 108
     @Test
107 109
     public void sortsSingleElement() {
108
-        List<Integer> expected = Collections.unmodifiableList(
109
-                Collections.singletonList(4)
110
+        BinarySearchTree<Character> binarySearchTree = new BinarySearchTree<>();
111
+        List<Character> expected = Collections.unmodifiableList(
112
+                Collections.singletonList('4')
110 113
         );
111 114
 
112
-        binarySearchTree.insert(4);
115
+        binarySearchTree.insert('4');
113 116
 
114
-        List<Integer> actual = binarySearchTree.getAsSortedList();
117
+        List<Character> actual = binarySearchTree.getAsSortedList();
115 118
         assertEquals(expected, actual);
116 119
     }
117 120
 
118 121
     @Ignore("Remove to run test")
119 122
     @Test
120 123
     public void sortsCollectionOfTwoIfSecondInsertedIsSmallerThanFirst() {
121
-        List<Integer> expected = Collections.unmodifiableList(
122
-                Arrays.asList(2, 4)
124
+        BinarySearchTree<String> binarySearchTree = new BinarySearchTree<>();
125
+        List<String> expected = Collections.unmodifiableList(
126
+                Arrays.asList("2", "4")
123 127
         );
124 128
 
125
-        binarySearchTree.insert(4);
126
-        binarySearchTree.insert(2);
129
+        binarySearchTree.insert("4");
130
+        binarySearchTree.insert("2");
127 131
 
128
-        List<Integer> actual = binarySearchTree.getAsSortedList();
132
+        List<String> actual = binarySearchTree.getAsSortedList();
129 133
         assertEquals(expected, actual);
130 134
     }
131 135
 
132 136
     @Ignore("Remove to run test")
133 137
     @Test
134 138
     public void sortsCollectionOfTwoIfSecondInsertedIsBiggerThanFirst() {
139
+        BinarySearchTree<Integer> binarySearchTree = new BinarySearchTree<>();
135 140
         List<Integer> expected = Collections.unmodifiableList(
136 141
                 Arrays.asList(4, 5)
137 142
         );
@@ -146,6 +151,7 @@ public class BinarySearchTreeTest {
146 151
     @Ignore("Remove to run test")
147 152
     @Test
148 153
     public void iteratesOverComplexTree() {
154
+        BinarySearchTree<Integer> binarySearchTree = new BinarySearchTree<>();
149 155
         List<Integer> expected = Collections.unmodifiableList(
150 156
                 Arrays.asList(1, 2, 3, 4, 5, 6, 7)
151 157
         );