require 'spec_helper'

describe Immutable::Set do
  describe '#count' do
    [
      [[], 0],
      [[1], 1],
      [[1, 2], 1],
      [[1, 2, 3], 2],
      [[1, 2, 3, 4], 2],
      [[1, 2, 3, 4, 5], 3],
    ].each do |values, expected|
      describe "on #{values.inspect}" do
        let(:set) { S[*values] }

        context 'with a block' do
          it "returns #{expected.inspect}" do
            set.count(&:odd?).should == expected
          end
        end

        context 'without a block' do
          it 'returns length' do
            set.count.should == set.length
          end
        end
      end
    end

    it 'works on large sets' do
      set = Immutable::Set.new(1..2000)
      set.count.should == 2000
      set.count(&:odd?).should == 1000
    end
  end
end
