require_relative '../../spec_helper'
require_relative 'fixtures/classes'

describe "Array#take" do
  it "returns the first specified number of elements" do
    [1, 2, 3].take(2).should == [1, 2]
  end

  it "returns all elements when the argument is greater than the Array size" do
    [1, 2].take(99).should == [1, 2]
  end

  it "returns all elements when the argument is less than the Array size" do
    [1, 2].take(4).should == [1, 2]
  end

  it "returns an empty Array when passed zero" do
    [1].take(0).should == []
  end

  it "returns an empty Array when called on an empty Array" do
    [].take(3).should == []
  end

  it "raises an ArgumentError when the argument is negative" do
    ->{ [1].take(-3) }.should.raise(ArgumentError)
  end

  it 'returns a Array instance for Array subclasses' do
    ArraySpecs::MyArray[1, 2, 3, 4, 5].take(1).should.instance_of?(Array)
  end
end
