blob: db42ef5c473eae1448819eafacd9aabd605aa6b8 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
require 'active_support/core_ext/module/aliasing'
class Range
begin
(1..2).step
# Range#step doesn't return an Enumerator
rescue LocalJumpError
# Return an array when step is called without a block.
def step_with_blockless(*args, &block)
if block_given?
step_without_blockless(*args, &block)
else
array = []
step_without_blockless(*args) { |step| array << step }
array
end
end
else
def step_with_blockless(*args, &block) #:nodoc:
if block_given?
step_without_blockless(*args, &block)
else
step_without_blockless(*args).to_a
end
end
end
alias_method_chain :step, :blockless
end
|
