Fixed yielded arrays of 1 element bug when using Enumerator::Yielder#…

…first
jruby · Jan 13, 2015 · cb25efe · cb25efe · nightscape · Apr 29, 2016
1 parent cd5683a
commit cb25efe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/core/src/main/java/org/jruby/RubyYielder.java b/core/src/main/java/org/jruby/RubyYielder.java
@@ -103,6 +103,10 @@ public IRubyObject yield(ThreadContext context, IRubyObject[]args) {
 
     @JRubyMethod(name = "<<", rest = true)
     public IRubyObject op_lshift(ThreadContext context, IRubyObject[]args) {
+        if (args.length == 1 &&
+                args[0] instanceof RubyArray &&
+                ((RubyArray) args[0]).getLength() == 1)
+            args[0] = RubyArray.newArray(context.runtime, args[0]);
         yield(context, args);
         return this;
     }