summaryrefslogtreecommitdiff
path: root/spec/ruby/library/matrix/columns_spec.rb
blob: 3095fdd7afb0c764d3850f6a719be6f356a8291a (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
require_relative '../../spec_helper'
require_relative 'fixtures/classes'
require 'matrix'

describe "Matrix.columns" do
  before :each do
    @a = [1, 2]
    @b = [3, 4]
    @m = Matrix.columns([@a, @b])
  end

  it "creates a Matrix from argument columns" do
    @m.should be_an_instance_of(Matrix)
    @m.column(0).to_a.should == @a
    @m.column(1).to_a.should == @b
  end

  it "accepts Vectors as argument columns" do
    m = Matrix.columns([Vector[*@a], Vector[*@b]])
    m.should == @m
    m.column(0).to_a.should == @a
    m.column(1).to_a.should == @b
  end

  it "handles empty matrices" do
    e = Matrix.columns([])
    e.row_size.should == 0
    e.column_size.should == 0
    e.should == Matrix[]

    v = Matrix.columns([[],[],[]])
    v.row_size.should == 0
    v.column_size.should == 3
    v.should == Matrix[[], [], []].transpose
  end

  describe "for a subclass of Matrix" do
    it "returns an instance of that subclass" do
      MatrixSub.columns([[1]]).should be_an_instance_of(MatrixSub)
    end
  end
end