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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
|
# -*- encoding: us-ascii -*-
require_relative '../../../spec_helper'
require_relative 'fixtures/classes'
describe "Enumerator::Lazy#drop" do
before :each do
@yieldsmixed = EnumeratorLazySpecs::YieldsMixed.new.to_enum.lazy
@eventsmixed = EnumeratorLazySpecs::EventsMixed.new.to_enum.lazy
ScratchPad.record []
end
after :each do
ScratchPad.clear
end
it "returns a new instance of Enumerator::Lazy" do
ret = @yieldsmixed.drop(1)
ret.should be_an_instance_of(Enumerator::Lazy)
ret.should_not equal(@yieldsmixed)
end
it "sets difference of given count with old size to new size" do
Enumerator::Lazy.new(Object.new, 100) {}.drop(20).size.should == 80
Enumerator::Lazy.new(Object.new, 100) {}.drop(200).size.should == 0
end
describe "when the returned lazy enumerator is evaluated by Enumerable#first" do
it "stops after specified times" do
(0..Float::INFINITY).lazy.drop(2).first(2).should == [2, 3]
@eventsmixed.drop(0).first(1)
ScratchPad.recorded.should == [:before_yield]
end
end
describe "on a nested Lazy" do
it "sets difference of given count with old size to new size" do
Enumerator::Lazy.new(Object.new, 100) {}.drop(20).drop(50).size.should == 30
Enumerator::Lazy.new(Object.new, 100) {}.drop(50).drop(20).size.should == 30
end
describe "when the returned lazy enumerator is evaluated by Enumerable#first" do
it "stops after specified times" do
(0..Float::INFINITY).lazy.drop(2).drop(2).first(2).should == [4, 5]
@eventsmixed.drop(0).drop(0).first(1)
ScratchPad.recorded.should == [:before_yield]
end
end
end
it "works with an infinite enumerable" do
s = 0..Float::INFINITY
s.lazy.drop(100).first(100).should ==
s.first(200).drop(100)
end
end
|