File: decomp_eig_gen.cpp

package info (click to toggle)
armadillo 1:9.200.7+dfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 7,500 kB
  • sloc: cpp: 137,626; makefile: 75; sh: 34
file content (126 lines) | stat: -rw-r--r-- 3,927 bytes parent folder | download | duplicates (2)
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Copyright 2015 Conrad Sanderson (http://conradsanderson.id.au)
// Copyright 2015 National ICT Australia (NICTA)
// 
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
// 
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ------------------------------------------------------------------------


#include <armadillo>
#include "catch.hpp"

using namespace arma;


TEST_CASE("decomp_eig_gen_1")
  {
  mat A = 
    "\
     0.061198   0.201990   0.019678  -0.493936  -0.126745;\
     0.437242   0.058956  -0.149362  -0.045465   0.296153;\
    -0.492474  -0.031309   0.314156   0.419733   0.068317;\
     0.336352   0.411541   0.458476  -0.393139  -0.135040;\
     0.239585  -0.428913  -0.406953  -0.291020  -0.353768;\
    ";
  
  cx_vec eigvals1 =
    {
    cx_double(-0.431507827005653, +0.336567219978257),
    cx_double(-0.431507827005653, -0.336567219978257),
    cx_double( 0.509611570246060, +0.000000000000000),
    cx_double( 0.020403541882623, +0.255686097698784),
    cx_double( 0.020403541882623, -0.255686097698784)
    };
  
  cx_vec eigvals2 = eig_gen(A);
  
  cx_vec eigvals3;
  bool status = eig_gen(eigvals3, A);
  
  cx_vec eigvals4;
  cx_mat eigvecs4;
  eig_gen(eigvals4, eigvecs4, A);
  
  cx_mat B = eigvecs4 * diagmat(eigvals4) * inv(eigvecs4);
  
  REQUIRE( status == true );
  REQUIRE( accu(abs(eigvals2 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(eigvals3 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(eigvals4 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(A        - B       )) == Approx(0.0).epsilon(0.0001) );
  }



TEST_CASE("decomp_eig_gen_2")
  {
  cx_mat A = 
    { 
    { cx_double( 0.111205, +0.074101), cx_double(-0.225872, -0.068474), cx_double(-0.192660, +0.236887), cx_double( 0.355204, -0.355735) },
    { cx_double( 0.119869, +0.217667), cx_double(-0.412722, +0.366157), cx_double( 0.069916, -0.222238), cx_double( 0.234987, -0.072355) },
    { cx_double( 0.003791, +0.183253), cx_double(-0.212887, -0.172758), cx_double( 0.168689, -0.393418), cx_double( 0.008795, -0.289654) },
    { cx_double(-0.331639, -0.166660), cx_double( 0.436969, -0.313498), cx_double(-0.431574, +0.017421), cx_double(-0.104165, +0.145246) }
    };

  cx_vec eigvals1 =
    {
    cx_double(-0.47418, +0.60377),
    cx_double( 0.15084, -0.44209),
    cx_double(-0.15790, -0.35629),
    cx_double( 0.24426, +0.38670)
    };

  cx_vec eigvals2 = eig_gen(A);
  
  cx_vec eigvals3;
  bool status = eig_gen(eigvals3, A);
  
  cx_vec eigvals4;
  cx_mat eigvecs4;
  eig_gen(eigvals4, eigvecs4, A);
  
  cx_mat B = eigvecs4 * diagmat(eigvals4) * inv(eigvecs4);
   
  REQUIRE( status == true );
  REQUIRE( accu(abs(eigvals2 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(eigvals3 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(eigvals4 - eigvals1)) == Approx(0.0).epsilon(0.0001) );
  REQUIRE( accu(abs(A        - B       )) == Approx(0.0).epsilon(0.0001) );
  }



TEST_CASE("decomp_eig_gen_3")
  {
  mat A(5,5,fill::randu);
  
  cx_vec eigvals(10,     fill::randu);
  cx_mat eigvecs(10, 10, fill::randu);
  
  A(0,0) = datum::inf;
  bool status = eig_gen(eigvals, eigvecs, A);
  
  REQUIRE( status         == false    );
  REQUIRE( eigvals.n_elem == uword(0) );
  REQUIRE( eigvecs.n_elem == uword(0) );
  }



TEST_CASE("decomp_eig_gen_4")
  {
  mat A(5,6,fill::randu);
  
  cx_vec eigvals;
  cx_mat eigvecs;
  
  REQUIRE_THROWS( eig_gen(eigvals, eigvecs, A) );
  }