array: Add chunkArray

Splits an array into the given maxSize chunks

Author: julienduchesne

array.libsonnet

@@ -52,4 +52,18 @@ local d = import 'github.com/jsonnet-libs/docsonnet/doc-util/main.libsonnet';
     for i in std.range(0, std.length(arr) - 1)
     if filter_func(i, arr[i])
   ],
+
+  '#chunkArray':: d.fn(
+    |||
+      `chunkArray` chunks an array into smaller arrays of the given max size.
+    |||,
+    [
+      d.arg('arr', d.T.array),
+      d.arg('maxSize', d.T.number),
+    ]
+  ),
+  chunkArray(arr, maxSize): [
+    arr[i * maxSize:std.min((i + 1) * maxSize, std.length(arr))]
+    for i in std.range(0, std.ceil(std.length(arr) / maxSize) - 1)
+  ],
 }

test/array_test.jsonnet

@@ -81,3 +81,32 @@ test.new(std.thisFile)
     expected=[0, 2, 4],
   )
 )
+
++ test.case.new(
+  name='chunkArray',
+  test=test.expect.eq(
+    actual=array.chunkArray(arr, maxSize=3),
+    expected=[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10]],
+  )
+)
++ test.case.new(
+  name='chunkArray - maxSize is 2',
+  test=test.expect.eq(
+    actual=array.chunkArray(arr, maxSize=2),
+    expected=[[0, 1], [2, 3], [4, 5], [6, 7], [8, 9], [10]],
+  )
+)
++ test.case.new(
+  name='chunkArray - maxSize is larger than array length',
+  test=test.expect.eq(
+    actual=array.chunkArray(arr, maxSize=100),
+    expected=[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]],
+  )
+)
++ test.case.new(
+  name='chunkArray - maxSize is 1',
+  test=test.expect.eq(
+    actual=array.chunkArray(arr, maxSize=1),
+    expected=[[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]],
+  )
+)