Working on the MySet

src/main/java/MyArrayList.java

@@ -17,6 +17,10 @@ public class MyArrayList<T> {
         this.size = size;
     }
 
+    public Object[] getMyArrayList() {
+        return myArrayList;
+    }
+
     public void add(T data) {
 
         Object[] tempArray1 = new Object[getSize() + 1];
@@ -89,7 +93,6 @@ public class MyArrayList<T> {
         decreaseSize();
     }
 
-
     public void set(int i, T data) throws ArrayIndexOutOfBoundsException {
         try {
             this.myArrayList[i] = data;
@@ -98,7 +101,6 @@ public class MyArrayList<T> {
         }
     }
 
-
     public void clear() {
         Object[] tempArr = new Object[INITIAL_SIZE];
         resetSize();
@@ -131,4 +133,7 @@ public class MyArrayList<T> {
         return count;
     }
 
+    public void setMyArrayList(Object[] myArrayList) {
+        this.myArrayList = myArrayList;
+    }
 }

src/main/java/MySet.java

@@ -1,7 +1,30 @@
-public class MySet<T> extends MyArrayList {
+public class MySet<T> extends MyArrayList<T> {
 
     public MySet() {
+
     }
 
 
+    @Override
+    public void add(T data){
+        // Sets cannot have duplicates
+        boolean hasItem = false;
+
+            for (Object o: getMyArrayList()) {
+                if (o.equals(data)) {
+                    hasItem = true;
+                    break;
+                }
+            }
+
+            if (!hasItem) {
+            Object[] tempArray1 = new Object[getSize() + 1];
+            copy(tempArray1, getMyArrayList());
+            tempArray1[getSize()] = data;
+            setMyArrayList(tempArray1);
+            setSize(getSize() + 1);
+
+
+        }
+    }
 }

src/test/java/MySetTest.java

@@ -1,11 +1,34 @@
+import org.junit.Assert;
 import org.junit.Before;
+import org.junit.Test;
 
 public class MySetTest {
 
-    MySet<Integer> myset;
+    MySet<Integer> mySet;
 
     @Before
     public void setUp() {
-        myset = new MySet<>();
+        mySet = new MySet<>();
+
+    }
+
+    @Test
+    public void testAddUnique() {
+        int expectedOutput = 1;
+        mySet.add(1);
+        Assert.assertEquals(expectedOutput, mySet.getMyArrayList()[mySet.getSize()-1]);
+    }
+
+    @Test
+    public void testAddNotUnique() {
+
+        int expectedSize = 2;
+        mySet.add(1);
+        mySet.add(2);
+        mySet.add(1);
+
+        Assert.assertEquals(expectedSize, mySet.getSize());
     }
+
+    
 }