summaryrefslogtreecommitdiff
path: root/spec/ruby/library/matrix/eigenvalue_decomposition/eigenvalues_spec.rb
blob: 8d47bed5e6748bde40637a5394402c2d45f98d79 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
require_relative '../../../spec_helper'
require 'matrix'

describe "Matrix::EigenvalueDecomposition#eigenvalues" do
  it "returns an array of complex eigenvalues for a rotation matrix" do
    Matrix[[ 1, 1],
           [-1, 1]].eigensystem.eigenvalues.sort_by{|v| v.imag}.should ==
    [ Complex(1, -1), Complex(1, 1)]
  end

  it "returns an array of real eigenvalues for a symetric matrix" do
    Matrix[[1, 2],
           [2, 1]].eigensystem.eigenvalues.sort.map!{|x| x.round(10)}.should ==
    [ -1, 3 ]
  end

  it "returns an array of real eigenvalues for a matrix" do
    Matrix[[14, 16],
           [-6, -6]].eigensystem.eigenvalues.sort.map!{|x| x.round(10)}.should ==
    [ 2, 6 ]
  end
end