File: take_last.rb

package info (click to toggle)
ruby-powerpack 0.1.1-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, buster, forky, sid, trixie
  • size: 492 kB
  • sloc: ruby: 819; makefile: 3
file content (20 lines) | stat: -rw-r--r-- 522 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
unless Enumerable.method_defined? :take_last
  module Enumerable
    # Take the last n elements of an enumerable.
    #
    # @param n [Fixnum] the number of elements to take
    # @return [Array] an array containing the requested elements
    #
    # @example
    #   [1, 2, 3].take_last(2) #=> [2, 3]
    #   [].take_last(5) #=> []
    def take_last(n)
      fail ArgumentError, 'attempt to take negative size' if n < 0

      ary = to_a

      return ary if n > ary.size
      ary[(ary.size - n)..-1]
    end
  end
end