Fix sized enumerators of Numeric#step with Infinity (#3683)

kachick · brixen · commit cf086d949ca7 · 2016-07-22T13:38:23.000-07:00
* Add specs for sized enumerator of Numeric#step with infinity limit

* Fix sized enumerator of Numeric#step with Infinity
diff --git a/core/numeric_mirror.rb b/core/numeric_mirror.rb
@@ -29,6 +29,14 @@ def step_size(limit, step, to, by)
         asc = values[3]
         is_float = values[4]
 
+        if limit == Float::INFINITY
+          if step == Float::INFINITY
+            return 1
+          else
+            return Float::INFINITY
+          end
+        end
+
         return Float::INFINITY if step == 0
 
         if is_float
diff --git a/spec/ruby/core/numeric/step_spec.rb b/spec/ruby/core/numeric/step_spec.rb
@@ -72,6 +72,14 @@
           it "should return infinity_value when step is 0.0" do
             1.step(to: 2, by: 0.0).size.should == infinity_value
           end
+
+          it "should return infinity_value when the limit is Float::INFINITY" do
+            1.step(to: Float::INFINITY, by: 42).size.should == infinity_value
+          end
+
+          it "should return 1 when the both limit and step are Float::INFINITY" do
+            1.step(to: Float::INFINITY, by: Float::INFINITY).size.should == 1
+          end
         end
       end
     end
@@ -130,7 +138,7 @@
       # a mix of positional and keyword arguments.
       # It's needed to test numeric_step behaviour with positional mixed with
       # keyword arguments.
-      @step_args = ->(*args) do 
+      @step_args = ->(*args) do
         if args.size == 2
           [args[0], {by: args[1]}]
         else
