Added another remove example (eugenp#2484)

* moved test file to correct location * added another remove example
jeffgrant · Aug 24, 2017 · 9b563d8 · 9b563d8
1 parent 154b59f
commit 9b563d8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java b/libraries/src/test/java/com/baeldung/commons/collections4/BagTests.java
@@ -34,9 +34,7 @@ public void whenAdd_thenRemoveFromBaseBag_thenContainsCorrect() {
 		Assert.assertFalse(baseBag.contains("lemon"));
 
 		Assert.assertTrue(baseBag.uniqueSet().contains("apple"));
-		Assert.assertFalse(baseBag.uniqueSet().contains("lemon"));
-		Assert.assertTrue(baseBag.uniqueSet().contains("lime"));
-
+
 		List<String> containList = new ArrayList<String>();
 		containList.add("apple");
 		containList.add("lemon");
@@ -56,11 +54,12 @@ public void whenAdd_thenRemoveFromBaseCollectionBag_thenContainsCorrect() {
 		baseCollectionBag.remove("lemon");
 		Assert.assertEquals(8, baseCollectionBag.size());
 		Assert.assertTrue(baseCollectionBag.contains("lemon"));
+
+		baseCollectionBag.remove("lemon",1);
+		Assert.assertEquals(7, baseCollectionBag.size());
 
 		Assert.assertTrue(baseBag.uniqueSet().contains("apple"));
-		Assert.assertTrue(baseBag.uniqueSet().contains("lemon"));
-		Assert.assertTrue(baseBag.uniqueSet().contains("lime"));
-
+
 		List<String> containList = new ArrayList<String>();
 		containList.add("apple");
 		containList.add("lemon");
@@ -81,5 +80,6 @@ public void whenAddtoTreeBag_thenRemove_thenContainsCorrect() {
 		treeBag.remove("apple");
 		Assert.assertEquals(9, treeBag.size());
 		Assert.assertEquals("banana", treeBag.first());
+
 	}
 }