iterator

src/main/java/MySet.java

@@ -1,4 +1,6 @@
-public class MySet<T> extends MyArrayList<T> {
+import java.util.Iterator;
+
+public class MySet<T> extends MyArrayList<T> implements Iterable<T> {
 
     public MySet() {
 
@@ -24,7 +26,40 @@ public class MySet<T> extends MyArrayList<T> {
             setMyArrayList(tempArray1);
             setSize(getSize() + 1);
 
+        }
+    }
+
+    public class SetIterator implements Iterator<T> {
+        int current = 0;
+
+        public boolean hasNext() {
+            if (current < getSize()) {
+                return true;
+            }
+            return false;
+        }
+
+        @Override
+        public T next() {
+            if (!hasNext()) {
+                throw new ArrayIndexOutOfBoundsException();
+            }
+
+            return (T) getMyArrayList()[current++];
+        }
+
+        @Override
+        public void remove() {
+        // removes current -1
+            if (current < getSize() - 1) {
+                MySet.this.remove(next());
+            }
 
         }
     }
+
+    @Override
+    public Iterator<T> iterator() {
+        return new SetIterator();
+    }
 }

src/test/java/MySetTest.java

@@ -30,5 +30,26 @@ public class MySetTest {
         Assert.assertEquals(expectedSize, mySet.getSize());
     }
 
-    
+    @Test
+    public void testContainsTrue() {
+
+        mySet.add(1);
+        mySet.add(2);
+        mySet.add(5);
+        mySet.add(10);
+
+        Assert.assertTrue(mySet.contains(2));
+    }
+
+    @Test
+    public void testContainsFalse() {
+        mySet.add(1);
+        mySet.add(2);
+        mySet.add(5);
+        mySet.add(10);
+
+        Assert.assertFalse(mySet.contains(3));
+    }
+
+
 }