diff --git a/lib/prime.rb b/lib/prime.rb index 3aa20f8..01a59c7 100644 --- a/lib/prime.rb +++ b/lib/prime.rb @@ -141,6 +141,18 @@ def each(ubound = nil, generator = EratosthenesGenerator.new, &block) generator.each(&block) end + # Return true if +obj+ is an Integer an is prime. Also returns + # true if +obj+ is a Module that is an ancestor of +Prime+. + def include?(obj) + case obj + when Integer + prime?(obj) + when Module + Module.instance_method(:include?).bind(Prime).call(obj) + else + false + end + end # Returns true if +value+ is a prime number, else returns false. # diff --git a/test/test_prime.rb b/test/test_prime.rb index 9c793da..1f9947f 100644 --- a/test/test_prime.rb +++ b/test/test_prime.rb @@ -27,6 +27,14 @@ def test_each assert_equal PRIMES, primes end + def test_include? + assert_equal(false, Prime.include?(nil)) + assert_equal(true, Prime.include?(3)) + assert_equal(false, Prime.include?(4)) + assert_equal(true, Prime.include?(Enumerable)) + assert_equal(false, Prime.include?(Comparable)) + end + def test_integer_each_prime primes = [] Integer.each_prime(1000) do |p|