require 'test/unit' class TestEnumerator < Test::Unit::TestCase def setup @obj = Object.new class << @obj include Enumerable def foo(*a) a.each {|x| yield x } end end end def enum_test obj i = 0 obj.map{|e| e }.sort end def test_iterators assert_equal [0, 1, 2], enum_test(3.times) assert_equal [:x, :y, :z], enum_test([:x, :y, :z].each) assert_equal [[:x, 1], [: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_loop e = 3.times i = 0 loop{ assert_equal(i, e.next) i += 1 } 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 def test_initialize assert_equal([1, 2, 3], @obj.to_enum(:foo, 1, 2, 3).to_a) assert_equal([1, 2, 3], Enumerator.new(@obj, :foo, 1, 2, 3).to_a) assert_equal([1, 2, 3], Enumerator.new { |y| i = 0; loop { y << (i += 1) } }.take(3)) assert_raise(ArgumentError) { Enumerator.new } end def test_initialize_copy assert_equal([1, 2, 3], @obj.to_enum(:foo, 1, 2, 3).dup.to_a) e = @obj.to_enum(:foo, 1, 2, 3) assert_nothing_raised { assert_equal(1, e.next) } assert_raise(TypeError) { e.dup } e = Enumerator.new { |y| i = 0; loop { y << (i += 1) } }.dup assert_nothing_raised { assert_equal(1, e.next) } assert_raise(TypeError) { e.dup } end def test_gc assert_nothing_raised do 1.times do foo = [1,2,3].to_enum GC.start end GC.start end end def test_slice assert_equal([[1,2,3],[4,5,6],[7,8,9],[10]], (1..10).each_slice(3).to_a) end def test_cons a = [[1,2,3], [2,3,4], [3,4,5], [4,5,6], [5,6,7], [6,7,8], [7,8,9], [8,9,10]] assert_equal(a, (1..10).each_cons(3).to_a) end def test_with_index assert_equal([[1,0],[2,1],[3,2]], @obj.to_enum(:foo, 1, 2, 3).with_index.to_a) assert_equal([[1,5],[2,6],[3,7]], @obj.to_enum(:foo, 1, 2, 3).with_index(5).to_a) end def test_with_object obj = [0, 1] ret = (1..10).each.with_object(obj) {|i, memo| memo[0] += i memo[1] *= i } assert_same(obj, ret) assert_equal([55, 3628800], ret) a = [2,5,2,1,5,3,4,2,1,0] obj = {} ret = a.delete_if.with_object(obj) {|i, seen| if seen.key?(i) true else seen[i] = true false end } assert_same(obj, ret) 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) assert_equal(2, e.next) e.rewind assert_equal(1, e.next) assert_equal(2, e.next) assert_equal(3, e.next) assert_raise(StopIteration) { e.next } end def test_peek a = [1] e = a.each assert_equal(1, e.peek) assert_equal(1, e.peek) assert_equal(1, e.next) assert_raise(StopIteration) { e.peek } assert_raise(StopIteration) { e.peek } end def test_peek_modify o = Object.new def o.each yield 1,2 end e = o.to_enum a = e.peek a << 3 assert_equal([1,2], e.peek) end def test_peek_values_modify o = Object.new def o.each yield 1,2 end e = o.to_enum a = e.peek_values a << 3 assert_equal([1,2], e.peek) end def test_next_after_stopiteration a = [1] e = a.each assert_equal(1, e.next) assert_raise(StopIteration) { e.next } assert_raise(StopIteration) { e.next } e.rewind assert_equal(1, e.next) assert_raise(StopIteration) { e.next } assert_raise(StopIteration) { e.next } end def test_stop_result a = [1] res = a.each {} e = a.each assert_equal(1, e.next) exc = assert_raise(StopIteration) { e.next } assert_equal(res, exc.result) end def test_next_values o = Object.new def o.each yield yield 1 yield 1, 2 end e = o.to_enum assert_equal(nil, e.next) assert_equal(1, e.next) assert_equal([1,2], e.next) e = o.to_enum assert_equal([], e.next_values) assert_equal([1], e.next_values) assert_equal([1,2], e.next_values) end def test_peek_values o = Object.new def o.each yield yield 1 yield 1, 2 end e = o.to_enum assert_equal(nil, e.peek) assert_equal(nil, e.next) assert_equal(1, e.peek) assert_equal(1, e.next) assert_equal([1,2], e.peek) assert_equal([1,2], e.next) e = o.to_enum assert_equal([], e.peek_values) assert_equal([], e.next_values) assert_equal([1], e.peek_values) assert_equal([1], e.next_values) assert_equal([1,2], e.peek_values) assert_equal([1,2], e.next_values) e = o.to_enum assert_equal([], e.peek_values) assert_equal(nil, e.next) assert_equal([1], e.peek_values) assert_equal(1, e.next) assert_equal([1,2], e.peek_values) assert_equal([1,2], e.next) e = o.to_enum assert_equal(nil, e.peek) assert_equal([], e.next_values) assert_equal(1, e.peek) assert_equal([1], e.next_values) assert_equal([1,2], e.peek) assert_equal([1,2], e.next_values) end def test_feed o = Object.new def o.each(ary) ary << yield ary << yield ary << yield end ary = [] e = o.to_enum(:each, ary) e.next e.feed 1 e.next e.feed 2 e.next e.feed 3 assert_raise(StopIteration) { e.next } assert_equal([1,2,3], ary) end def test_feed_mixed o = Object.new def o.each(ary) ary << yield ary << yield ary << yield end ary = [] e = o.to_enum(:each, ary) e.next e.feed 1 e.next e.next e.feed 3 assert_raise(StopIteration) { e.next } assert_equal([1,nil,3], ary) end def test_feed_twice o = Object.new def o.each(ary) ary << yield ary << yield ary << yield end ary = [] e = o.to_enum(:each, ary) e.feed 1 assert_raise(TypeError) { e.feed 2 } end def test_feed_before_first_next o = Object.new def o.each(ary) ary << yield ary << yield ary << yield end ary = [] e = o.to_enum(:each, ary) e.feed 1 e.next e.next assert_equal([1], ary) end def test_rewind_clear_feed o = Object.new def o.each(ary) ary << yield ary << yield ary << yield end ary = [] e = o.to_enum(:each, ary) e.next e.feed 1 e.next e.feed 2 e.rewind e.next e.next assert_equal([1,nil], ary) end def test_feed_yielder x = nil e = Enumerator.new {|y| x = y.yield; 10 } e.next e.feed 100 exc = assert_raise(StopIteration) { e.next } assert_equal(100, x) assert_equal(10, exc.result) end def test_inspect e = (0..10).each_cons(2) assert_equal("#", e.inspect) e = Enumerator.new {|y| x = y.yield; 10 } assert_match(/\A#/, e.inspect) a = [] e = a.each_with_object(a) a << e assert_equal("#]:each_with_object([#])>", e.inspect) end def test_generator # note: Enumerator::Generator is a class just for internal g = Enumerator::Generator.new {|y| y << 1 << 2 << 3; :foo } g2 = g.dup a = [] assert_equal(:foo, g.each {|x| a << x }) assert_equal([1, 2, 3], a) a = [] assert_equal(:foo, g2.each {|x| a << x }) assert_equal([1, 2, 3], a) end def test_yielder # note: Enumerator::Yielder is a class just for internal a = [] y = Enumerator::Yielder.new {|x| a << x } assert_equal(y, y << 1 << 2 << 3) assert_equal([1, 2, 3], a) a = [] y = Enumerator::Yielder.new {|x| a << x } assert_equal([1], y.yield(1)) assert_equal([1, 2], y.yield(2)) assert_equal([1, 2, 3], y.yield(3)) assert_raise(LocalJumpError) { Enumerator::Yielder.new } end end