summaryrefslogtreecommitdiff
path: root/spec/ruby/library/matrix/vector/each2_spec.rb
blob: bdd6966472ef271b71fba20690da9efd7f58a1bf (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
require_relative '../../../spec_helper'

ruby_version_is ""..."3.1" do
  require 'matrix'

  describe "Vector.each2" do
    before :all do
      @v = Vector[1, 2, 3]
      @v2 = Vector[4, 5, 6]
    end

    it "requires one argument" do
      -> { @v.each2(@v2, @v2){} }.should raise_error(ArgumentError)
      -> { @v.each2(){} }.should raise_error(ArgumentError)
    end

    describe "given one argument" do
      it "accepts an Array argument" do
        a = []
        @v.each2([7, 8, 9]){|x, y| a << x << y}
        a.should == [1, 7, 2, 8, 3, 9]
      end

      it "raises a DimensionMismatch error if the Vector size is different" do
        -> { @v.each2(Vector[1,2]){}     }.should raise_error(Vector::ErrDimensionMismatch)
        -> { @v.each2(Vector[1,2,3,4]){} }.should raise_error(Vector::ErrDimensionMismatch)
      end

      it "yields arguments in sequence" do
        a = []
        @v.each2(@v2){|first, second| a << [first, second]}
        a.should == [[1, 4], [2, 5], [3, 6]]
      end

      it "yield arguments in pairs" do
        a = []
        @v.each2(@v2){|*pair| a << pair}
        a.should == [[1, 4], [2, 5], [3, 6]]
      end

      it "returns self when given a block" do
        @v.each2(@v2){}.should equal(@v)
      end

      it "returns an enumerator if no block given" do
        enum = @v.each2(@v2)
        enum.should be_an_instance_of(Enumerator)
        enum.to_a.should == [[1, 4], [2, 5], [3, 6]]
      end
    end
  end
end