Contents

require File.join(File.expand_path(File.dirname(__FILE__)), 'testutilities')
require File.expand_path(File.join(File.dirname(__FILE__), "../lib/amp/support/generator"))

class FibonacciGeneratorTester < Generator
  def generator_loop
    a, b = 0, 1
    while true
      yield_gen b
      a, b = b, a + b
    end
  end
end

class TestGenerator < AmpTestCase
  def setup
    @generator = FibonacciGeneratorTester.new
  end
  
  def test_next
    assert_equal 1, @generator.next
    assert_equal 1, @generator.next
    assert_equal 2, @generator.next
    assert_equal 3, @generator.next
    assert_equal 5, @generator.next
  end
  
  def test_reset
    10.times { @generator.next }
    #reset it, first num should be 1
    @generator.reset
    assert_equal 1, @generator.next
  end
  
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
amp-0.5.3	test/test_generator.rb