require "spec_helper"
require "hamster/vector"

describe Hamster::Vector do
  describe "#repeated_permutation" do
    let(:vector) { V[1,2,3,4] }

    context "without a block" do
      context "and without argument" do
        it "returns an Enumerator of all repeated permutations" do
          expect(vector.repeated_permutation.class).to be(Enumerator)
          expect(vector.repeated_permutation.to_a.sort).to eql(vector.to_a.repeated_permutation(4).to_a.sort)
        end
      end

      context "with an integral argument" do
        it "returns an Enumerator of all repeated permutations of the given length" do
          expect(vector.repeated_permutation(2).class).to be(Enumerator)
          expect(vector.repeated_permutation(2).to_a.sort).to eql(vector.to_a.repeated_permutation(2).to_a.sort)
          expect(vector.repeated_permutation(3).class).to be(Enumerator)
          expect(vector.repeated_permutation(3).to_a.sort).to eql(vector.to_a.repeated_permutation(3).to_a.sort)
        end
      end
    end

    context "with a block" do
      it "returns self" do
        expect(vector.repeated_permutation {}).to be(vector)
      end

      context "on an empty vector" do
        it "yields the empty permutation" do
          yielded = []
          V.empty.repeated_permutation { |obj| yielded << obj }
          expect(yielded).to eql([[]])
        end
      end

      context "with an argument of zero" do
        it "yields the empty permutation" do
          yielded = []
          vector.repeated_permutation(0) { |obj| yielded << obj }
          expect(yielded).to eql([[]])
        end
      end

      context "with no argument" do
        it "yields all repeated permutations" do
          yielded = []
          V[1,2,3].repeated_permutation { |obj| yielded << obj }
          expect(yielded.sort).to eql([[1,1,1], [1,1,2], [1,1,3], [1,2,1], [1,2,2],
            [1,2,3], [1,3,1], [1,3,2], [1,3,3], [2,1,1], [2,1,2], [2,1,3], [2,2,1],
            [2,2,2], [2,2,3], [2,3,1], [2,3,2], [2,3,3], [3,1,1], [3,1,2], [3,1,3],
            [3,2,1], [3,2,2], [3,2,3], [3,3,1], [3,3,2], [3,3,3]])
        end
      end

      context "with a positive integral argument" do
        it "yields all repeated permutations of the given length" do
          yielded = []
          vector.repeated_permutation(2) { |obj| yielded << obj }
          expect(yielded.sort).to eql([[1,1], [1,2], [1,3], [1,4], [2,1], [2,2], [2,3], [2,4],
            [3,1], [3,2], [3,3], [3,4], [4,1], [4,2], [4,3], [4,4]])
        end
      end
    end

    it "handles duplicate elements correctly" do
    expect(V[10,11,10].repeated_permutation(2).sort).to eql([[10, 10], [10, 10],
      [10, 10], [10, 10], [10, 11], [10, 11], [11, 10], [11, 10], [11, 11]])
    end

    it "allows permutations larger than the number of elements" do
      expect(V[1,2].repeated_permutation(3).sort).to eql(
        [[1, 1, 1], [1, 1, 2], [1, 2, 1],
         [1, 2, 2], [2, 1, 1], [2, 1, 2],
         [2, 2, 1], [2, 2, 2]])
    end

    it "leaves the original unmodified" do
      vector.repeated_permutation(2) {}
      expect(vector).to eql(V[1,2,3,4])
    end

    it "behaves like Array#repeated_permutation" do
      10.times do
        array  = rand(8).times.map { rand(10000) }
        vector = V.new(array)
        perm_size = array.size == 0 ? 0 : rand(array.size)
        expect(array.repeated_permutation(perm_size).to_a).to eq(vector.repeated_permutation(perm_size).to_a)
      end
    end
  end
end