Merge pull request eugenp#3257 from shouvikbhattacharya/BAEL-1433

evandor · web-flow · commit 617c2ffdba31 · 2017-12-18T06:50:12.000+01:00
BAEL-1433: Guide to TreeSet in Java
diff --git a/core-java/src/test/java/com/baeldung/collection/WhenUsingTreeSet.java b/core-java/src/test/java/com/baeldung/collection/WhenUsingTreeSet.java
@@ -26,7 +26,6 @@ public void setId(Integer id) {
 
         @Override
         public String toString() {
-            // TODO Auto-generated method stub
             return id.toString();
         }
     }
@@ -189,8 +188,14 @@ public void whenUsingSubSet_shouldReturnSubSetElements() {
         treeSet.add(5);
         treeSet.add(6);
 
+        Set<Integer> expectedSet = new TreeSet<>();
+        expectedSet.add(2);
+        expectedSet.add(3);
+        expectedSet.add(4);
+        expectedSet.add(5);
+
         Set<Integer> subSet = treeSet.subSet(2, 6);
-        System.out.println(subSet);
+        Assert.assertEquals(expectedSet, subSet);
     }
 
     @Test
