fixes iterator tests

Author: amejiarosario

src/data-structures/sets/array-set.js

@@ -1,7 +1,7 @@
 /**
  * Set implemented with our HashMap to have sublinear times on all operations
  */
-class HashMapSet {
+class ArraySet {
   /**
    * Initialize Set using a HashMap
    *
@@ -62,7 +62,14 @@ class HashMapSet {
   entries() {
     return this.array;
   }
+
+  /**
+   * Make this class iterable
+   */
+  [Symbol.iterator]() {
+    return this.entries()[Symbol.iterator]();
+  }
 }
 
-module.exports = HashMapSet;
+module.exports = ArraySet;
 

src/data-structures/sets/hash-set-1.js

@@ -48,7 +48,14 @@ class HashMapSet {
    * Return all values on the set as an array
    */
   entries() {
-    return Array.from(this.hashMap.keys());
+    return Array.from(this);
+  }
+
+  /**
+   * Make this class iterable
+   */
+  [Symbol.iterator]() {
+    return this.hashMap.keys()[Symbol.iterator]();
   }
 }
 

src/data-structures/sets/hash-set.js

@@ -47,7 +47,14 @@ class HashSet {
    * Return all values on the set as an array
    */
   entries() {
-    return Array.from(this.hashMap.keys());
+    return Array.from(this);
+  }
+
+  /**
+   * Make this class iterable
+   */
+  [Symbol.iterator]() {
+    return this.hashMap.keys();
   }
 }
 

src/data-structures/sets/set.spec.js

@@ -2,7 +2,7 @@ const ArraySet = require('./array-set');
 const HashSet = require('./hash-set');
 const HashMapSet = require('./hash-set-1');
 
-[ArraySet, HashSet, HashMapSet].forEach((MySet) => {
+[HashSet, HashMapSet, ArraySet].forEach((MySet) => {
   describe(`Set (${MySet.name})`, () => {
     let set;
 
@@ -66,7 +66,7 @@ const HashMapSet = require('./hash-set-1');
       expect(set.entries()).toEqual([1, 2, 3]);
     });
 
-    xit('should return an iterable', () => {
+    it('should return an iterable', () => {
       set = new MySet([1, 2, 3, 1]);
       expect(Array.from(set)).toEqual([1, 2, 3]);
     });