spec/ruby/library/prime/integer/each_prime_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13

require File.expand_path('../../../../spec_helper', __FILE__)
require 'prime'

describe "Integer.each_prime" do
  it "is transferred to Prime.each" do
    Prime.should_receive(:each).with(100).and_yield(2).and_yield(3).and_yield(5)
    yielded = []
    Integer.each_prime(100) do |prime|
      yielded << prime
    end
    yielded.should == [2,3,5]
  end
end