From a96fbe6ac48093a9c3bef44e3c7f0dd5c6782f90 Mon Sep 17 00:00:00 2001
From: marcandre <marcandre@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Tue, 20 Oct 2015 03:19:44 +0000
Subject: * lib/prime.rb: Add basic argument checking to Prime.prime?     [Bug
 #11606]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@52201 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/prime.rb | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'lib/prime.rb')

diff --git a/lib/prime.rb b/lib/prime.rb
index d99a51a967..c79b0cfd5f 100644
--- a/lib/prime.rb
+++ b/lib/prime.rb
@@ -144,6 +144,8 @@ class Prime
   # +value+:: an arbitrary integer to be checked.
   # +generator+:: optional. A pseudo-prime generator.
   def prime?(value, generator = Prime::Generator23.new)
+    raise ArgumentError, "Expected a prime generator, got #{generator}" unless generator.respond_to? :each
+    raise ArgumentError, "Expected an integer, got #{value}" unless value.respond_to?(:integer?) && value.integer?
     return false if value < 2
     generator.each do |num|
       q,r = value.divmod num
-- 
cgit v1.2.3