From 07bd65ae65ba1092191f5a702ce3c15aaca2ab5f Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" Date: Fri, 21 Mar 2014 13:35:39 +0900 Subject: flat_map to support multiple values; ref #1902 --- mrbgems/mruby-enum-ext/mrblib/enum.rb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'mrbgems/mruby-enum-ext/mrblib') diff --git a/mrbgems/mruby-enum-ext/mrblib/enum.rb b/mrbgems/mruby-enum-ext/mrblib/enum.rb index 6413198d4..d0e125850 100644 --- a/mrbgems/mruby-enum-ext/mrblib/enum.rb +++ b/mrbgems/mruby-enum-ext/mrblib/enum.rb @@ -256,10 +256,10 @@ module Enumerable return to_enum :flat_map unless block_given? ary = [] - self.each do |e| - e2 = block.call(e) + self.each do |*e| + e2 = block.call(*e) if e2.respond_to? :each - e2.each { |e3| ary.push(e3) } + e2.each {|e3| ary.push(e3) } else ary.push(e2) end -- cgit v1.2.3