File: permutation_spec.rb

package info (click to toggle)
ruby-hamster 3.0.0-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 1,932 kB
  • sloc: ruby: 16,915; makefile: 4
file content (92 lines) | stat: -rw-r--r-- 3,015 bytes parent folder | download | duplicates (2)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
require "spec_helper"
require "hamster/vector"

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

    context "without a block or arguments" do
      it "returns an Enumerator of all permutations" do
        vector.permutation.class.should be(Enumerator)
        vector.permutation.to_a.should eql(vector.to_a.permutation.to_a)
      end
    end

    context "without a block, but with integral argument" do
      it "returns an Enumerator of all permutations of given length" do
        vector.permutation(2).class.should be(Enumerator)
        vector.permutation(2).to_a.should eql(vector.to_a.permutation(2).to_a)
        vector.permutation(3).class.should be(Enumerator)
        vector.permutation(3).to_a.should eql(vector.to_a.permutation(3).to_a)
      end
    end

    context "with a block" do
      it "returns self" do
        vector.permutation {}.should be(vector)
      end

      context "and no argument" do
        it "yields all permutations" do
          yielded = []
          vector.permutation { |obj| yielded << obj }
          yielded.sort.should eql([[1,2,3,4], [1,2,4,3], [1,3,2,4], [1,3,4,2],
            [1,4,2,3], [1,4,3,2], [2,1,3,4], [2,1,4,3], [2,3,1,4], [2,3,4,1],
            [2,4,1,3], [2,4,3,1], [3,1,2,4], [3,1,4,2], [3,2,1,4], [3,2,4,1],
            [3,4,1,2], [3,4,2,1], [4,1,2,3], [4,1,3,2], [4,2,1,3], [4,2,3,1],
            [4,3,1,2], [4,3,2,1]])
        end
      end

      context "and an integral argument" do
        it "yields all permutations of the given length" do
          yielded = []
          vector.permutation(2) { |obj| yielded << obj }
          yielded.sort.should eql([[1,2], [1,3], [1,4], [2,1], [2,3], [2,4], [3,1],
            [3,2], [3,4], [4,1], [4,2], [4,3]])
        end
      end
    end

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

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

    context "with a length greater than the size of the vector" do
      it "yields no permutations" do
        vector.permutation(5) { |obj| fail }
      end
    end

    it "handles duplicate elements correctly" do
      V[1,2,3,1].permutation(2).sort.should eql([[1,1], [1,1], [1,2], [1,2],
        [1,3], [1,3], [2,1],[2,1],[2,3], [3,1],[3,1],[3,2]])
    end

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

    it "behaves like Array#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)
        array.permutation(perm_size).to_a.should == vector.permutation(perm_size).to_a
      end
    end
  end
end