From 6b323c45cc3731d4447412e3b14faef1f3ba890d Mon Sep 17 00:00:00 2001
From: shouvikbhattacharya <shouvikbhattacharya@outlook.com>
Date: Sat, 16 Dec 2017 21:28:03 +0530
Subject: [PATCH] BAEL-1433: Minor changes.

---
 .../java/com/baeldung/collection/WhenUsingTreeSet.java   | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/core-java/src/test/java/com/baeldung/collection/WhenUsingTreeSet.java b/core-java/src/test/java/com/baeldung/collection/WhenUsingTreeSet.java
index 887f6eba1d..52d5b08813 100644
--- 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 class WhenUsingTreeSet {
 
         @Override
         public String toString() {
-            // TODO Auto-generated method stub
             return id.toString();
         }
     }
@@ -189,8 +188,14 @@ public class WhenUsingTreeSet {
         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