creates a flat_map function

Author: EvandroLG

src/array/init.lua

@@ -641,6 +641,17 @@ array = {
     end
 
     return output
+  end,
+
+  -- Returns a new array-like table by applying a given callback to each element of the table, and then flattening the result by one level
+  -- @param obj {table}
+  -- @param callback {function}
+  -- @return {table}
+  flat_map = function(obj, callback)
+    utils.raises_error(array, obj, 'flat_map')
+    local mapped = array.map(obj, callback)
+
+    return array.flat(mapped)
   end
 }
 

test.lua

@@ -425,3 +425,12 @@ test('chunk', function(a)
     }
   )
 end)
+
+test('flat_map', function(a)
+  a.deep_equal(
+    array.flat_map({ 1, 2, 3 }, function(value)
+      return { value, value * 2 }
+    end),
+    { 1, 2, 2, 4, 3, 6 }
+  )
+end)