From 5791afc9fe42e0bbfb6fc96dbf7fdb68a79fd2af Mon Sep 17 00:00:00 2001
From: ko1 <ko1@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Mon, 20 Aug 2007 18:58:32 +0000
Subject: * enumerator.c (next_i): fix to return with Fiber#yield at   the end
 of each block.  [ruby-dev:31470] * enumerator.c (enumerator_next_p): call
 init_next if not   initialized.  [ruby-dev:31514] *
 test/ruby/test_enumerator.rb: add tests for Enumerator.

git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@13120 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/ruby/test_enumerator.rb | 48 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 test/ruby/test_enumerator.rb

(limited to 'test/ruby')

diff --git a/test/ruby/test_enumerator.rb b/test/ruby/test_enumerator.rb
new file mode 100644
index 000000000..6ff8f8201
--- /dev/null
+++ b/test/ruby/test_enumerator.rb
@@ -0,0 +1,48 @@
+require 'test/unit'
+
+class TestEnumerator < Test::Unit::TestCase
+  def enum_test obj
+    i = 0
+    obj.map{|e|
+      [i+=1, e]
+    }
+  end
+
+  def test_iterators
+    assert_equal [[1, 0], [2, 1], [3, 2]], enum_test(3.times)
+    assert_equal [[1, :x], [2, :y], [3, :z]], enum_test([:x, :y, :z].each)
+    assert_equal [[1, [:x, 1]], [2, [:y, 2]]], enum_test({:x=>1, :y=>2})
+  end
+
+  ## Enumerator as Iterator
+
+  def test_next
+    e = 3.times
+    3.times{|i|
+      assert_equal i, e.next
+    }
+    assert_raise(StopIteration){e.next}
+  end
+
+  def test_next?
+    e = 3.times
+    assert_equal true, e.next?
+    3.times{|i|
+      assert_equal true, e.next?
+      assert_equal i, e.next
+    }
+    assert_equal false, e.next?
+  end
+
+  def test_nested_itaration
+    def (o = Object.new).each
+      yield :ok1
+      yield [:ok2, :x].each.next
+    end
+    e = o.to_enum
+    assert_equal :ok1, e.next
+    assert_equal :ok2, e.next
+    assert_raise(StopIteration){e.next}
+  end
+end
+
-- 
cgit