diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/matrix.rb | 3 | ||||
-rw-r--r-- | lib/prime.rb | 6 |
2 files changed, 4 insertions, 5 deletions
diff --git a/lib/matrix.rb b/lib/matrix.rb index 7f338bb07e..34602f5c40 100644 --- a/lib/matrix.rb +++ b/lib/matrix.rb @@ -514,12 +514,11 @@ class Matrix # # => prints the numbers 1 to 4 # Matrix[ [1,2], [3,4] ].each(:strict_lower).to_a # => [3] # - def each(which = :all) # :yield: e + def each(which = :all, &block) # :yield: e return to_enum :each, which unless block_given? last = column_count - 1 case which when :all - block = Proc.new @rows.each do |row| row.each(&block) end diff --git a/lib/prime.rb b/lib/prime.rb index 9f6db448be..3aa20f8def 100644 --- a/lib/prime.rb +++ b/lib/prime.rb @@ -283,9 +283,9 @@ class Prime end # see +Enumerator+#with_index. - def with_index(offset = 0) - return enum_for(:with_index, offset) { Float::INFINITY } unless block_given? - return each_with_index(&proc) if offset == 0 + def with_index(offset = 0, &block) + return enum_for(:with_index, offset) { Float::INFINITY } unless block + return each_with_index(&block) if offset == 0 each do |prime| yield prime, offset |