diff --git a/src/test/java/MyStackTest.java b/src/test/java/MyStackTest.java index b49e82d29d1b970e72e9c794fba030af42a7f84f..562c9d052e58e121c7fa44ee2317f5e339d28292 100644 --- a/src/test/java/MyStackTest.java +++ b/src/test/java/MyStackTest.java @@ -17,6 +17,12 @@ public class MyStackTest { public void tearDown() { } + private void pushDummies(int numberOfDummies) { + for (int i=0; i<numberOfDummies; i++) { + stack.push(new Object()); + } + } + @Test public void testEmptyStack() { assertEquals(0, stack.size()); @@ -24,7 +30,13 @@ public class MyStackTest { @Test public void testPushOneElement() { - stack.push(new Object()); + pushDummies(1); assertEquals(1, stack.size()); } + + @Test + public void testPushThreeElements() { + pushDummies(3); + assertEquals(3, stack.size()); + } } \ No newline at end of file