From 0b35e9e5ddb329dac027d72851db015a738852d7 Mon Sep 17 00:00:00 2001 From: shugo Date: Wed, 14 Mar 2012 10:29:25 +0000 Subject: * enumerator.c (lazy_take): add Enumerable::Lazy#take. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35018 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- test/ruby/test_lazy_enumerator.rb | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'test') diff --git a/test/ruby/test_lazy_enumerator.rb b/test/ruby/test_lazy_enumerator.rb index 9e50cd3a69..3adaaa32f3 100644 --- a/test/ruby/test_lazy_enumerator.rb +++ b/test/ruby/test_lazy_enumerator.rb @@ -143,4 +143,12 @@ class TestLazyEnumerator < Test::Unit::TestCase assert_equal(["a", 1], a.lazy.zip("a".."c") {|x, y| [y, x]}.first) assert_equal(1, a.current) end + + def test_take + a = Step.new(1..3) + assert_equal(1, a.take(2).first) + assert_equal(2, a.current) + assert_equal(1, a.lazy.take(2).first) + assert_equal(1, a.current) + end end -- cgit v1.2.3