128 lines
4.9 KiB
C++
128 lines
4.9 KiB
C++
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// 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("fn_symmat_1")
|
|
{
|
|
mat A =
|
|
"\
|
|
0.061198 0.201990 0.019678 -0.493936 -0.126745 0.051408;\
|
|
0.437242 0.058956 -0.149362 -0.045465 0.296153 0.035437;\
|
|
-0.492474 -0.031309 0.314156 0.419733 0.068317 -0.454499;\
|
|
0.336352 0.411541 0.458476 -0.393139 -0.135040 0.373833;\
|
|
0.239585 -0.428913 -0.406953 -0.291020 -0.353768 0.258704;\
|
|
";
|
|
|
|
mat B = symmatu( A(0,0,size(5,5)) );
|
|
mat C = symmatl( A(0,0,size(5,5)) );
|
|
|
|
mat BB =
|
|
"\
|
|
0.061198 0.201990 0.019678 -0.493936 -0.126745;\
|
|
0.201990 0.058956 -0.149362 -0.045465 0.296153;\
|
|
0.019678 -0.149362 0.314156 0.419733 0.068317;\
|
|
-0.493936 -0.045465 0.419733 -0.393139 -0.135040;\
|
|
-0.126745 0.296153 0.068317 -0.135040 -0.353768;\
|
|
";
|
|
|
|
mat CC =
|
|
"\
|
|
0.061198 0.437242 -0.492474 0.336352 0.239585;\
|
|
0.437242 0.058956 -0.031309 0.411541 -0.428913;\
|
|
-0.492474 -0.031309 0.314156 0.458476 -0.406953;\
|
|
0.336352 0.411541 0.458476 -0.393139 -0.291020;\
|
|
0.239585 -0.428913 -0.406953 -0.291020 -0.353768;\
|
|
";
|
|
|
|
REQUIRE( accu(abs( B - BB )) == Approx(0.0).margin(0.001) );
|
|
REQUIRE( accu(abs( C - CC )) == Approx(0.0).margin(0.001) );
|
|
|
|
mat X;
|
|
REQUIRE_THROWS( X = symmatu(A) ); // symmatu() and symmatl() currently handle only square matrices
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE("fn_symmat_2")
|
|
{
|
|
mat A =
|
|
"\
|
|
0.061198 0.201990 0.019678 -0.493936 -0.126745 0.051408;\
|
|
0.437242 0.058956 -0.149362 -0.045465 0.296153 0.035437;\
|
|
-0.492474 -0.031309 0.314156 0.419733 0.068317 -0.454499;\
|
|
0.336352 0.411541 0.458476 -0.393139 -0.135040 0.373833;\
|
|
0.239585 -0.428913 -0.406953 -0.291020 -0.353768 0.258704;\
|
|
";
|
|
|
|
cx_mat B1 = symmatu( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ) );
|
|
cx_mat C1 = symmatl( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ) );
|
|
|
|
cx_mat B2 = symmatu( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ), true );
|
|
cx_mat C2 = symmatl( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ), true );
|
|
|
|
cx_mat D = symmatu( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ), false );
|
|
cx_mat E = symmatl( cx_mat( A(0,0,size(3,3)), A(0,3,size(3,3)) ), false );
|
|
|
|
cx_mat BB =
|
|
{
|
|
{ cx_double( 0.06120, -0.49394), cx_double( 0.20199, -0.12674), cx_double( 0.01968, +0.05141) },
|
|
{ cx_double( 0.20199, +0.12674), cx_double( 0.05896, +0.29615), cx_double(-0.14936, +0.03544) },
|
|
{ cx_double( 0.01968, -0.05141), cx_double(-0.14936, -0.03544), cx_double( 0.31416, -0.45450) }
|
|
};
|
|
|
|
cx_mat CC =
|
|
{
|
|
{ cx_double( 0.06120, -0.49394), cx_double( 0.43724, +0.04546), cx_double(-0.49247, -0.41973) },
|
|
{ cx_double( 0.43724, -0.04546), cx_double( 0.05896, +0.29615), cx_double(-0.03131, -0.06832) },
|
|
{ cx_double(-0.49247, +0.41973), cx_double(-0.03131, +0.06832), cx_double( 0.31416, -0.45450) }
|
|
};
|
|
|
|
cx_mat DD =
|
|
{
|
|
{ cx_double( 0.06120, -0.49394), cx_double( 0.20199, -0.12674), cx_double( 0.01968, +0.05141) },
|
|
{ cx_double( 0.20199, -0.12674), cx_double( 0.05896, +0.29615), cx_double(-0.14936, +0.03544) },
|
|
{ cx_double( 0.01968, +0.05141), cx_double(-0.14936, +0.03544), cx_double( 0.31416, -0.45450) }
|
|
};
|
|
|
|
cx_mat EE =
|
|
{
|
|
{ cx_double( 0.06120, -0.49394), cx_double( 0.43724, -0.04546), cx_double(-0.49247, +0.41973) },
|
|
{ cx_double( 0.43724, -0.04546), cx_double( 0.05896, +0.29615), cx_double(-0.03131, +0.06832) },
|
|
{ cx_double(-0.49247, +0.41973), cx_double(-0.03131, +0.06832), cx_double( 0.31416, -0.45450) }
|
|
};
|
|
|
|
REQUIRE( accu(abs( B1 - BB )) == Approx(0.0).margin(0.0001) );
|
|
REQUIRE( accu(abs( C1 - CC )) == Approx(0.0).margin(0.0001) );
|
|
|
|
REQUIRE( accu(abs( B2 - BB )) == Approx(0.0).margin(0.0001) );
|
|
REQUIRE( accu(abs( C2 - CC )) == Approx(0.0).margin(0.0001) );
|
|
|
|
REQUIRE( accu(abs( D - DD )) == Approx(0.0).margin(0.0001) );
|
|
REQUIRE( accu(abs( E - EE )) == Approx(0.0).margin(0.0001) );
|
|
|
|
cx_mat X;
|
|
REQUIRE_THROWS( X = symmatu( cx_mat(A(0,0,size(2,3)), A(0,3,size(2,3))) ) ); // symmatu() and symmatl() currently handle only square matrices
|
|
}
|
|
|
|
|