summaryrefslogtreecommitdiffhomepage
path: root/mrbgems/mruby-enum-ext/test/enum.rb
diff options
context:
space:
mode:
Diffstat (limited to 'mrbgems/mruby-enum-ext/test/enum.rb')
-rw-r--r--mrbgems/mruby-enum-ext/test/enum.rb10
1 files changed, 10 insertions, 0 deletions
diff --git a/mrbgems/mruby-enum-ext/test/enum.rb b/mrbgems/mruby-enum-ext/test/enum.rb
index 9ab508a11..0f7708db2 100644
--- a/mrbgems/mruby-enum-ext/test/enum.rb
+++ b/mrbgems/mruby-enum-ext/test/enum.rb
@@ -13,3 +13,13 @@ assert("Enumrable#drop_while") do
assert_equal a.drop_while {|i| i < 3 }, [3, 4, 5, 0]
end
+assert("Enumrable#take") do
+ a = [1, 2, 3, 4, 5, 0]
+ assert_equal a.take(3), [1, 2, 3]
+end
+
+assert("Enumrable#take_while") do
+ a = [1, 2, 3, 4, 5, 0]
+ assert_equal a.take_while {|i| i < 3 }, [1, 2]
+end
+