From a96cf7a7a5d984dcc3b87f297c3f55ff31fd308d Mon Sep 17 00:00:00 2001 From: knu Date: Tue, 3 Jun 2008 12:43:45 +0000 Subject: * enumerator.c (enumerator_with_memo): New method: with_memo(). git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@16795 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- test/ruby/test_enumerator.rb | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'test') diff --git a/test/ruby/test_enumerator.rb b/test/ruby/test_enumerator.rb index 31382b03b1..8645b7acf7 100644 --- a/test/ruby/test_enumerator.rb +++ b/test/ruby/test_enumerator.rb @@ -91,6 +91,25 @@ class TestEnumerator < Test::Unit::TestCase assert_equal([[1,0],[2,1],[3,2]], @obj.to_enum(:foo, 1, 2, 3).with_index.to_a) end + def test_with_memo + r = 1..10 + assert_equal([55, 3628800], (1..10).each.with_memo([0,1]) {|i, memo| + memo[0] += i + memo[1] *= i + }) + + a = [2,5,2,1,5,3,4,2,1,0] + a.delete_if.with_memo({}) {|i, seen| + if seen.key?(i) + true + else + seen[i] = true + false + end + } + assert_equal([2, 5, 1, 3, 4, 0], a) + end + def test_next_rewind e = @obj.to_enum(:foo, 1, 2, 3) assert_equal(1, e.next) -- cgit v1.2.3