NSArray.chunk()

NSArray+YOLO.m

@@ -7,6 +7,16 @@ - (NSSet *)set {
     return [NSSet setWithArray:self];
 }
 
+- (NSArray *(^)(NSUInteger))chunk {
+    return ^(NSUInteger size){
+        id aa = [NSMutableArray new];
+        const int n = self.count / size;
+        for (int x = 0; x < n; ++x)
+            [aa addObject:self.slice(x*size, size)];
+        return aa;
+    };
+}
+
 - (NSArray *(^)(id (^)(id o)))pmap {
     return ^NSArray *(id (^block)(id))
     {

README.markdown

@@ -461,6 +461,15 @@ code.
 
 `has` was chosen over `contains` or `includes` because it is short and clear.
 
+###NSArray.chunk()
+```objc
+id rv = @[@1, @2, @3, @4].chunk(2)
+
+// rv => @[@[@1, @2], @[@3, @4]]
+```
+
+Chunks your array into an array of chunk-size arrays.
+
 ###NSArray.partition()
 ```objc
 id rv = @[@"A", @"B", @"AA"].partition(^(NSString *s){

YOLO.h

@@ -47,6 +47,7 @@
 @end
 
 @interface NSArray (YOLO)
+- (NSArray *(^)(NSUInteger))chunk;
 - (BOOL)empty;
 - (BOOL (^)(id o))has;
 - (NSArray *(^)(id (^)(id o)))pmap;

tests

@@ -491,6 +491,12 @@ int main() {
     STAssertEqualObjects(aa, bb, @"");
 }
 
+- (void)testChunk {
+    id aa = @[@1, @2, @3, @4].chunk(2);
+    id bb = @[@[@1, @2], @[@3, @4]];
+    STAssertEqualObjects(aa, bb, @"");
+}
+
 - (void)testPartition {
     id aa = @[@1, @2, @3, @4].partition(^(NSNumber *n){
         return n.intValue % 2 == 0;