extra-iterator 0.9.0 → 0.9.1

package/dist/index.js

@@ -137,7 +137,7 @@ export class ExtraIterator extends Iterator {
      * @example ExtraIterator.from([[1, 2], [3, 4]]).flatten().toArray() // returns [1, 2, 3, 4]
      */
     flatten() {
-        return this.flatMap(value => value);
+        return this.flatMap(value => Array.isArray(value) ? new ExtraIterator(value).flatten() : [value]);
     }
     /**
      * Creates a new iterator that yields the values of this iterator, but won't yield any duplicates.

package/dist/index.test.js

@@ -47,8 +47,8 @@ describe('ExtraIterator', () => {
         expect(iterator.toArray()).toEqual([3, 4]);
     });
     it('should flatten nested iterables', () => {
-        const iterator = ExtraIterator.from([[1, 2], [3, 4]]).flatten();
-        expect(iterator.toArray()).toEqual([1, 2, 3, 4]);
+        const iterator = ExtraIterator.from([0, [1, [2, [3, 4]]], [5, [6]], 7]).flatten();
+        expect(iterator.toArray()).toEqual([0, 1, 2, 3, 4, 5, 6, 7]);
     });
     it('should return the first value', () => {
         const iterator = ExtraIterator.from([1, 2, 3]);

package/package.json

@@ -3,7 +3,7 @@
   "description": "An extension of the Iterator class with additional utility helper functions.",
   "author": "Leonardo Raele <leonardoraele@gmail.com>",
   "license": "MIT",
-  "version": "0.9.0",
+  "version": "0.9.1",
   "type": "module",
   "exports": {
     ".": "./dist/index.js",