summaryrefslogtreecommitdiff
path: root/spec/ruby/core/array/product_spec.rb
blob: 7d810b6196ff2463c6d70bf23eaccadcdae2fbef (plain)
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
require_relative '../../spec_helper'
require_relative 'fixtures/classes'

describe "Array#product" do
  it "returns converted arguments using :to_ary" do
    lambda{ [1].product(2..3) }.should raise_error(TypeError)
    ar = ArraySpecs::ArrayConvertible.new(2,3)
    [1].product(ar).should == [[1,2],[1,3]]
    ar.called.should == :to_ary
  end

  it "returns the expected result" do
    [1,2].product([3,4,5],[6,8]).should == [[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
                                            [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8]]
  end

  it "has no required argument" do
    [1,2].product.should == [[1],[2]]
  end

  it "returns an empty array when the argument is an empty array" do
    [1, 2].product([]).should == []
  end

  it "does not attempt to produce an unreasonable number of products" do
    a = (0..100).to_a
    lambda do
      a.product(a, a, a, a, a, a, a, a, a, a)
    end.should raise_error(RangeError)
  end

  describe "when given a block" do
    it "yields all combinations in turn" do
      acc = []
      [1,2].product([3,4,5],[6,8]){|array| acc << array}
      acc.should == [[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
                     [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8]]

      acc = []
      [1,2].product([3,4,5],[],[6,8]){|array| acc << array}
      acc.should be_empty
    end

    it "returns self" do
      a = [1, 2, 3].freeze

      a.product([1, 2]) { |p| p.first }.should == a
    end

    it "will ignore unreasonable numbers of products and yield anyway" do
      a = (0..100).to_a
      lambda do
        a.product(a, a, a, a, a, a, a, a, a, a)
      end.should raise_error(RangeError)
    end
  end

  describe "when given an empty block" do
    it "returns self" do
      arr = [1,2]
      arr.product([3,4,5],[6,8]){}.should equal(arr)
      arr = []
      arr.product([3,4,5],[6,8]){}.should equal(arr)
      arr = [1,2]
      arr.product([]){}.should equal(arr)
    end
  end
end