* lib/prime.rb: Return sized enumerators.

  Patch by Kenichi Kamiya [GH-931]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@50857 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 6 insertions, 2 deletions

diff --git a/lib/prime.rb b/lib/prime.rb
index 1eed28d914..c9575f6a2a 100644
--- a/lib/prime.rb
+++ b/lib/prime.rb
@@ -270,7 +270,7 @@ class Prime
 
     # see +Enumerator+#with_index.
     def with_index(offset = 0)
-      return enum_for(:with_index, offset) unless block_given?
+      return enum_for(:with_index, offset) { Float::INFINITY } unless block_given?
       return each_with_index(&proc) if offset == 0
 
       each do |prime|
@@ -281,11 +281,15 @@ class Prime
 
     # see +Enumerator+#with_object.
     def with_object(obj)
-      return enum_for(:with_object, obj) unless block_given?
+      return enum_for(:with_object, obj) { Float::INFINITY } unless block_given?
       each do |prime|
         yield prime, obj
       end
     end
+
+    def size
+      Float::INFINITY
+    end
   end
 
   # An implementation of +PseudoPrimeGenerator+.