Added flat

calmm-js · Feb 13, 2018 · 7c5abc9 · 7c5abc9
1 parent f8680b5
commit 7c5abc9
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/src/partial.lenses.js b/src/partial.lenses.js
@@ -781,6 +781,13 @@ export function compose() {
   }
 }
 
+export function flat() {
+  const r = [flatten]
+  for (let i = 0, n = arguments.length; i < n; ++i)
+    r.push(arguments[i], flatten)
+  return r
+}
+
 // Recursing
 
 export function lazy(o2o) {

diff --git a/test/tests.js b/test/tests.js
@@ -232,6 +232,7 @@ describe('arities', () => {
     find: 1,
     findWith: 1,
     first: undefined,
+    flat: 0,
     flatten: 4,
     foldTraversalLens: 2,
     foldl: 4,
@@ -1705,6 +1706,21 @@ describe('L.pointer', () => {
   })
 })
 
+describe('L.flat', () => {
+  testEq(
+    `L.modify(
+      L.flat(
+        'a',
+        'b',
+        'c'
+      ),
+      R.negate,
+      [{a: [[{b: {c: 1}}], [{b: [{c: 2}]}]]}, {a: {b: {c: [[[3]]]}}}]
+    )`,
+    [{a: [[{b: {c: -1}}], [{b: [{c: -2}]}]]}, {a: {b: {c: [[[-3]]]}}}]
+  )
+})
+
 if (process.env.NODE_ENV !== 'production') {
   describe('debug', () => {
     testThrows(`X.set(-1, 0, 0)`)

diff --git a/test/types.js b/test/types.js
@@ -93,6 +93,7 @@ export const traverse = T.fn(
 // Nesting
 
 export const compose = T.fnVarN(0, T_optic, T_optic)
+export const flat = T.fnVarN(0, T_optic, T_optic)
 
 // Recursing