Merge pull request #2655 from tmtm/fix-numeric-step

Fix: Numeric#step infinite loop.

1 files changed, 11 insertions, 3 deletions

diff --git a/mrblib/numeric.rb b/mrblib/numeric.rb
index 5be3c90fc..1f44a2c81 100644
--- a/mrblib/numeric.rb
+++ b/mrblib/numeric.rb
@@ -101,12 +101,20 @@ module Integral
   # incremented by +step+ (default 1).
   #
   def step(num, step=1, &block)
+    raise ArgumentError, "step can't be 0" if step == 0
     return to_enum(:step, num, step) unless block_given?
 
     i = if num.kind_of? Float then self.to_f else self end
-    while(i <= num)
-      block.call(i)
-      i += step
+    if step > 0
+      while(i <= num)
+        block.call(i)
+        i += step
+      end
+    else
+      while(i >= num)
+        block.call(i)
+        i += step
+      end
     end
     self
   end