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
|
require File.expand_path('../../../../spec_helper', __FILE__)
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
lambda { @v.each2(@v2, @v2){} }.should raise_error(ArgumentError)
lambda { @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
lambda { @v.each2(Vector[1,2]){} }.should raise_error(Vector::ErrDimensionMismatch)
lambda { @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
|
