modify core method that return Enumerator object if not block given

mruby · Mar 14, 2014 · bc63518 · bc63518
1 parent 29964cf
commit bc63518
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mrblib/array.rb b/mrblib/array.rb
@@ -31,6 +31,8 @@ def each(&block)
   #
   # ISO 15.2.12.5.11
   def each_index(&block)
+    return to_enum :each_index unless block_given?
+
     idx = 0
     while(idx < length)
       block.call(idx)
@@ -46,6 +48,8 @@ def each_index(&block)
   #
   # ISO 15.2.12.5.7
   def collect!(&block)
+    return to_enum :collect! unless block_given?
+
     self.each_index{|idx|
       self[idx] = block.call(self[idx])
     }