add Enumerable#drop

2 files changed, 31 insertions, 0 deletions

diff --git a/mrbgems/mruby-enum-ext/mrblib/enum.rb b/mrbgems/mruby-enum-ext/mrblib/enum.rb
index af0b5bd10..9c77577ca 100644
--- a/mrbgems/mruby-enum-ext/mrblib/enum.rb
+++ b/mrbgems/mruby-enum-ext/mrblib/enum.rb
@@ -1,3 +1,24 @@
+##
+# Enumerable
+#
 module Enumerable
+  ##
+  # call-seq:
+  #    enum.drop(n)               -> array
+  #
+  # Drops first n elements from <i>enum</i>, and returns rest elements
+  # in an array.
+  #
+  #    a = [1, 2, 3, 4, 5, 0]
+  #    a.drop(3)             #=> [4, 5, 0]
+
+  def drop(n)
+    raise TypeError, "expected Integer for 1st argument" unless n.kind_of? Integer
+    raise ArgumentError, "attempt to drop negative size" if n < 0
+
+    ary = []
+    self.each {|e| n == 0 ? ary << e : n -= 1 }
+    ary
+  end
   
 end
diff --git a/mrbgems/mruby-enum-ext/test/enum.rb b/mrbgems/mruby-enum-ext/test/enum.rb
index e69de29bb..76ab08f37 100644
--- a/mrbgems/mruby-enum-ext/test/enum.rb
+++ b/mrbgems/mruby-enum-ext/test/enum.rb
@@ -0,0 +1,10 @@
+##
+# Enumerable(Ext) Test
+
+assert("Enumrable#drop") do
+  a = [1, 2, 3, 4, 5, 0]
+
+  assert_equal a.drop(3), [4, 5, 0]
+  assert_equal a.drop(6), []
+end
+