jruby · Mar 9, 2015 · Mar 9, 2015
Showing with 18 additions and 9 deletions.

+0 −9 spec/truffle/tags/core/array/cycle_tags.txt

+18 −0 truffle/src/main/ruby/core/rubinius/common/array.rb
diff --git a/spec/truffle/tags/core/array/cycle_tags.txt b/spec/truffle/tags/core/array/cycle_tags.txt
@@ -1,11 +1,2 @@
 fails:Array#cycle iterates indefinitely when not passed a count
 fails:Array#cycle iterates indefinitely when passed nil
-fails:Array#cycle does not rescue StopIteration when not passed a count
-fails:Array#cycle does not rescue StopIteration when passed a count
-fails:Array#cycle iterates the array Integer(count) times when passed a Float count
-fails:Array#cycle calls #to_int to convert count to an Integer
-fails:Array#cycle raises a TypeError if #to_int does not return an Integer
-fails:Array#cycle raises a TypeError if passed a String
-fails:Array#cycle raises a TypeError if passed an Object
-fails:Array#cycle raises a TypeError if passed true
-fails:Array#cycle raises a TypeError if passed false
diff --git a/truffle/src/main/ruby/core/rubinius/common/array.rb b/truffle/src/main/ruby/core/rubinius/common/array.rb
@@ -391,6 +391,24 @@ def bsearch
     nil
   end
 
+  def cycle(n=nil)
+    return to_enum(:cycle, n) unless block_given?
+    return nil if empty?
+
+    # Don't use nil? because, historically, lame code has overridden that method
+    if n.equal? nil
+      while true
+        each { |x| yield x }
+      end
+    else
+      n = Rubinius::Type.coerce_to_collection_index n
+      n.times do
+        each { |x| yield x }
+      end
+    end
+    nil
+  end
+
   def flatten(level=-1)
     level = Rubinius::Type.coerce_to_collection_index level
     return self.dup if level == 0