Enumerable#chunk should behave as a proc or lambda. not normal.

headius · Mar 4, 2021 · 30d5761 · 30d5761
1 parent bb5a9ab
commit 30d5761
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/src/main/java/org/jruby/RubyEnumerable.java b/core/src/main/java/org/jruby/RubyEnumerable.java
@@ -1989,7 +1989,8 @@ public static IRubyObject chunk(ThreadContext context, IRubyObject self, final B
 
         IRubyObject enumerator = runtime.getEnumerator().allocate();
         enumerator.getInternalVariables().setInternalVariable("chunk_enumerable", self);
-        enumerator.getInternalVariables().setInternalVariable("chunk_categorize", RubyProc.newProc(runtime, block, block.type));
+        enumerator.getInternalVariables().setInternalVariable("chunk_categorize",
+                RubyProc.newProc(runtime, block, block.type == Block.Type.LAMBDA ? block.type : Block.Type.PROC));
 
         Helpers.invoke(context, enumerator, "initialize",
                 CallBlock.newCallClosure(context, self, Signature.ONE_ARGUMENT,