1 files changed, 35 insertions, 21 deletions

diff --git a/test/jrummikub/model/TableTest.java b/test/jrummikub/model/TableTest.java
index 36d2a22..44605e2 100644
--- a/test/jrummikub/model/TableTest.java
+++ b/test/jrummikub/model/TableTest.java
@@ -14,70 +14,84 @@ import org.junit.Test;
 
 public class TableTest {
 	Table testTable;
-	
+
 	@Before
 	public void setup() {
 		testTable = new Table();
 	}
-	
+
 	@Test
 	public void testIsValid() {
-		testTable.drop(new StoneSet(Arrays.asList(new Stone(RED), new Stone(BLACK),
-				new Stone(1, BLACK))), new Position(0,0));
-		testTable.drop(new StoneSet(Arrays.asList(new Stone(1, RED), new Stone(2, RED),
-				new Stone(3, RED))), new Position(0,0));
+		testTable.drop(
+				new StoneSet(Arrays.asList(new Stone(RED), new Stone(BLACK),
+						new Stone(1, BLACK))), new Position(0, 0));
+		testTable.drop(
+				new StoneSet(Arrays.asList(new Stone(1, RED),
+						new Stone(2, RED), new Stone(3, RED))), new Position(0,
+						0));
 		assertTrue(testTable.isValid());
 
-		testTable.drop(new StoneSet(Arrays.asList(new Stone(5, RED))), new Position(0,0));
+		testTable.drop(new StoneSet(Arrays.asList(new Stone(5, RED))),
+				new Position(0, 0));
 		assertFalse(testTable.isValid());
 	}
-	
+
 	@Test
 	public void testEmptyIsValid() {
 		assertTrue(testTable.isValid());
 	}
-	
+
 	@Test
 	@SuppressWarnings("unused")
 	public void testPickUpStoneGroup() {
 		Stone targetStone = new Stone(BLACK);
-		testTable.drop(new StoneSet(Arrays.asList(new Stone(RED), targetStone,
-				new Stone(1, BLACK))), new Position(0,0));
+		testTable.drop(
+				new StoneSet(Arrays.asList(new Stone(RED), targetStone,
+						new Stone(1, BLACK))), new Position(0, 0));
 		assertTrue(testTable.isValid());
 		testTable.pickUpStone(targetStone);
 		assertFalse(testTable.isValid());
-		
+
 		int counter = 0;
 		for (Object i : testTable) {
 			counter++;
 		}
 		assertEquals(1, counter);
 	}
-	
+
+	@Test
+	public void testPickLonelyStone() {
+		Stone targetStone = new Stone(BLACK);
+		testTable.drop(new StoneSet(targetStone), new Position(0, 0));
+		testTable.pickUpStone(targetStone);
+		assertEquals(0, testTable.getSize());
+	}
+
 	@Test
 	@SuppressWarnings("unused")
 	public void testPickUpStoneRun() {
 		Stone targetStone = new Stone(BLACK);
-		testTable.drop(new StoneSet(Arrays.asList(new Stone(1, RED), targetStone,
-				new Stone(3, RED))), new Position(0,0));
+		testTable.drop(
+				new StoneSet(Arrays.asList(new Stone(1, RED), targetStone,
+						new Stone(3, RED))), new Position(0, 0));
 		assertTrue(testTable.isValid());
 		testTable.pickUpStone(targetStone);
 		assertFalse(testTable.isValid());
-		
+
 		int counter = 0;
 		for (Object i : testTable) {
 			counter++;
 		}
 		assertEquals(2, counter);
 	}
-	
+
 	@Test
 	public void testFindSet() {
 		Stone targetStone = new Stone(BLACK);
-		StoneSet testSet = new StoneSet(Arrays.asList(new Stone(1, RED), targetStone,
-				new Stone(3, RED)));
-		testTable.drop(testSet, new Position(0,0));
-		
+		StoneSet testSet = new StoneSet(Arrays.asList(new Stone(1, RED),
+				targetStone, new Stone(3, RED)));
+		testTable.drop(testSet, new Position(0, 0));
+
 		StoneSet foundSet = testTable.findStoneSet(targetStone);
 		assertSame(testSet, foundSet);
 	}