190 lines
4.2 KiB
C++
190 lines
4.2 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("init_misc_1")
|
|
{
|
|
const uword n_rows = 5;
|
|
const uword n_cols = 6;
|
|
|
|
mat A(5,6, fill::zeros);
|
|
|
|
for(uword row=0; row<n_rows; ++row)
|
|
for(uword col=0; col<n_cols; ++col)
|
|
{
|
|
REQUIRE(A(row,col) == 0);
|
|
}
|
|
|
|
mat B(5,6, fill::ones);
|
|
|
|
for(uword row=0; row<n_rows; ++row)
|
|
for(uword col=0; col<n_cols; ++col)
|
|
{
|
|
REQUIRE(B(row,col) == 1);
|
|
}
|
|
|
|
mat C(2,3);
|
|
|
|
C(0,0) = 0;
|
|
C(1,0) = 1;
|
|
|
|
C(0,1) = 2;
|
|
C(1,1) = 3;
|
|
|
|
C(0,2) = 4;
|
|
C(1,2) = 5;
|
|
|
|
REQUIRE(C(0,0) == 0);
|
|
REQUIRE(C(1,0) == 1);
|
|
|
|
REQUIRE(C(0,1) == 2);
|
|
REQUIRE(C(1,1) == 3);
|
|
|
|
REQUIRE(C(0,2) == 4);
|
|
REQUIRE(C(1,2) == 5);
|
|
|
|
mat D = { {0, 2, 4},
|
|
{1, 3, 5} };
|
|
|
|
REQUIRE(D(0,0) == 0);
|
|
REQUIRE(D(1,0) == 1);
|
|
|
|
REQUIRE(D(0,1) == 2);
|
|
REQUIRE(D(1,1) == 3);
|
|
|
|
REQUIRE(D(0,2) == 4);
|
|
REQUIRE(D(1,2) == 5);
|
|
|
|
mat E;
|
|
E << 0 << 2 << 4 << endr
|
|
<< 1 << 3 << 5 << endr;
|
|
|
|
REQUIRE(E(0,0) == 0);
|
|
REQUIRE(E(1,0) == 1);
|
|
|
|
REQUIRE(E(0,1) == 2);
|
|
REQUIRE(E(1,1) == 3);
|
|
|
|
REQUIRE(E(0,2) == 4);
|
|
REQUIRE(E(1,2) == 5);
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE("init_misc_2")
|
|
{
|
|
mat A =
|
|
{
|
|
{ -0.78838, 0.69298, 0.41084, 0.90142 },
|
|
{ 0.49345, -0.12020, 0.78987, 0.53124 },
|
|
{ 0.73573, 0.52104, -0.22263, 0.40163 }
|
|
};
|
|
|
|
mat B =
|
|
"\
|
|
-0.78838, 0.69298, 0.41084, 0.90142;\
|
|
0.49345, -0.12020, 0.78987, 0.53124;\
|
|
0.73573, 0.52104, -0.22263, 0.40163;\
|
|
";
|
|
|
|
mat C =
|
|
"\
|
|
-0.78838 0.69298 0.41084 0.90142;\
|
|
0.49345 -0.12020 0.78987 0.53124;\
|
|
0.73573 0.52104 -0.22263 0.40163;\
|
|
";
|
|
|
|
// NOTE: use of << for initialisation is deprecated
|
|
mat D;
|
|
D << -0.78838 << 0.69298 << 0.41084 << 0.90142 << endr
|
|
<< 0.49345 << -0.12020 << 0.78987 << 0.53124 << endr
|
|
<< 0.73573 << 0.52104 << -0.22263 << 0.40163 << endr;
|
|
|
|
REQUIRE( A.n_rows == 3 );
|
|
REQUIRE( A.n_cols == 4 );
|
|
|
|
REQUIRE( A(0,0) == Approx(-0.78838) );
|
|
REQUIRE( A(1,0) == Approx( 0.49345) );
|
|
REQUIRE( A(2,0) == Approx( 0.73573) );
|
|
REQUIRE( A(0,1) == Approx( 0.69298) );
|
|
REQUIRE( A(1,1) == Approx(-0.12020) );
|
|
REQUIRE( A(2,1) == Approx( 0.52104) );
|
|
REQUIRE( A(0,2) == Approx( 0.41084) );
|
|
REQUIRE( A(1,2) == Approx( 0.78987) );
|
|
REQUIRE( A(2,2) == Approx(-0.22263) );
|
|
REQUIRE( A(0,3) == Approx( 0.90142) );
|
|
REQUIRE( A(1,3) == Approx( 0.53124) );
|
|
REQUIRE( A(2,3) == Approx( 0.40163) );
|
|
|
|
REQUIRE( accu(abs(A-B)) == Approx(0.0).margin(0.001) );
|
|
REQUIRE( accu(abs(A-C)) == Approx(0.0).margin(0.001) );
|
|
REQUIRE( accu(abs(A-D)) == Approx(0.0).margin(0.001) );
|
|
}
|
|
|
|
|
|
|
|
TEST_CASE("init_misc_3")
|
|
{
|
|
const uword n_rows = 5;
|
|
const uword n_cols = 6;
|
|
|
|
cx_mat A(5,6, fill::zeros);
|
|
|
|
for(uword row=0; row<n_rows; ++row)
|
|
for(uword col=0; col<n_cols; ++col)
|
|
{
|
|
REQUIRE(A(row,col) == cx_double(0,0));
|
|
}
|
|
|
|
cx_mat B(5,6, fill::ones);
|
|
|
|
for(uword row=0; row<5; ++row)
|
|
for(uword col=0; col<6; ++col)
|
|
{
|
|
REQUIRE(B(row,col) == cx_double(1,0));
|
|
}
|
|
|
|
cx_mat C(2,3);
|
|
|
|
C(0,0) = cx_double(0.0, 1.0);
|
|
C(1,0) = cx_double(1.0, 2.0);
|
|
|
|
C(0,1) = cx_double(3.0, 4.0);
|
|
C(1,1) = cx_double(5.0, 6.0);
|
|
|
|
C(0,2) = cx_double(7.0, 8.0);
|
|
C(1,2) = cx_double(9.0, 10.0);
|
|
|
|
REQUIRE(C(0,0) == cx_double(0.0, 1.0));
|
|
REQUIRE(C(1,0) == cx_double(1.0, 2.0));
|
|
|
|
REQUIRE(C(0,1) == cx_double(3.0, 4.0));
|
|
REQUIRE(C(1,1) == cx_double(5.0, 6.0));
|
|
|
|
REQUIRE(C(0,2) == cx_double(7.0, 8.0));
|
|
REQUIRE(C(1,2) == cx_double(9.0, 10.0));
|
|
}
|
|
|
|
|
|
|