Shouldn't call each method if size is 0

mruby · Nov 24, 2016 · 5134734 · 5134734
1 parent 9e2faec
commit 5134734
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mrbgems/mruby-enum-ext/mrblib/enum.rb b/mrbgems/mruby-enum-ext/mrblib/enum.rb
@@ -61,9 +61,11 @@ def take(n)
     i = n.to_int
     raise ArgumentError, "attempt to take negative size" if i < 0
     ary = []
+    return ary if i == 0
     self.each do |*val|
-      break if ary.size >= n
       ary << val.__svalue
+      i -= 1
+      break if i == 0
     end
     ary
   end