* enumerator.c (lazy_cycle): add Enumerable::Lazy#cycle.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35028 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 10 insertions, 0 deletions

diff --git a/test/ruby/test_lazy_enumerator.rb b/test/ruby/test_lazy_enumerator.rb
index 6b4d49a041..1c0096f88c 100644
--- a/test/ruby/test_lazy_enumerator.rb
+++ b/test/ruby/test_lazy_enumerator.rb
@@ -184,6 +184,16 @@ class TestLazyEnumerator < Test::Unit::TestCase
     assert_equal([4, 5], (1..Float::INFINITY).lazy.drop(3).take(2).to_a)
   end
 
+  def test_cycle
+    a = Step.new(1..3)
+    assert_equal("1", a.cycle(2).map(&:to_s).first)
+    assert_equal(3, a.current)
+    assert_equal("1", a.lazy.cycle(2).map(&:to_s).first)
+    assert_equal(1, a.current)
+    assert_equal("1", a.lazy.cycle(2, &:to_s).first)
+    assert_equal(1, a.current)
+  end
+
   def test_force
     assert_equal([1, 2, 3], (1..Float::INFINITY).lazy.take(3).force)
   end