summaryrefslogtreecommitdiff
path: root/spec/ruby/library/matrix/shared/trace.rb
blob: 57b89863f8b561d652a3de290bcacca08928e96e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
require 'matrix'

describe :trace, shared: true do
  it "returns the sum of diagonal elements in a square Matrix" do
    Matrix[[7,6], [3,9]].trace.should == 16
  end

  it "returns the sum of diagonal elements in a rectangular Matrix" do
    ->{ Matrix[[1,2,3], [4,5,6]].trace}.should raise_error(Matrix::ErrDimensionMismatch)
  end

end