sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
47443b309da85e6799109bff6d969f7245b0f97c
Aurora/test/Function2D_Test.cpp

201 lines
9.8 KiB
C++
Raw Normal View History

init
2023-04-18 13:31:14 +08:00
#include <gtest/gtest.h>
Add repmat and interp unittest.
2023-04-21 17:12:53 +08:00
#include <vector>
Arrange Test directory.
2023-04-23 09:30:47 +08:00
#include "TestUtility.h"
init
2023-04-18 13:31:14 +08:00
#include "Matrix.h"
#include "Function.h"
Matrix class update.
2023-04-19 15:54:52 +08:00
#include "Function1D.h"
Fix matrix operator bug on - and /. 3
2023-04-21 17:04:09 +08:00
#include "Function2D.h"
Add repmat and interp unittest.
2023-04-21 17:12:53 +08:00
#include "Function3D.h"
init
2023-04-18 13:31:14 +08:00
Arrange Test directory.
2023-04-23 09:30:47 +08:00
class Function2D_Test: public ::testing::Test{
init
2023-04-18 13:31:14 +08:00
protected:
static void SetUpFunctionTester(){
}
static void TearDownTestCase(){
}
void SetUp(){
}
void TearDown(){
}
};
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, immse){
Fix matrix operator bug on - and /. 3
2023-04-21 17:04:09 +08:00
double *dataA = new double[9]{1,2,3,4,5,6,7,8,9};
double *dataB = new double[9]{10,20,30,40,50,40,30,20,10};
Aurora::Matrix A = Aurora::Matrix::fromRawData(dataA,3,3);
Aurora::Matrix B = Aurora::Matrix::fromRawData(dataB,3,3);
double v = immse(A,B);
EXPECT_DOUBLE_EQ(fourDecimalRound(v),698.3333)<<" immse error;";
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, inv){
Fix matrix operator bug on - and /. 3
2023-04-21 17:04:09 +08:00
//默认是column major排布数据
double *dataA=new double[9]{2, 0, 2, 2, 3, 0, 3, 3, 3};
double *dataB=new double[9]{1, 1, 1, 1, 1, 1, 1, 1, 1};
Aurora::Matrix A = Aurora::Matrix::fromRawData(dataA,3,3);
Aurora::Matrix B = Aurora::Matrix::fromRawData(dataB,3,3);
auto invA = Aurora::inv(A);
double* result = invA.getData();
EXPECT_DOUBLE_EQ(0.75, fourDecimalRound(result[0]));
EXPECT_DOUBLE_EQ(0.5, fourDecimalRound(result[1]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[2]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[3]));
EXPECT_DOUBLE_EQ(.0, fourDecimalRound(result[4]));
EXPECT_DOUBLE_EQ(0.3333, fourDecimalRound(result[5]));
EXPECT_DOUBLE_EQ(-0.25, fourDecimalRound(result[6]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[7]));
EXPECT_DOUBLE_EQ(0.5, fourDecimalRound(result[8]));
invA = Aurora::inv(A*B);
result = invA.getData();
EXPECT_DOUBLE_EQ(0.75, fourDecimalRound(result[0]));
EXPECT_DOUBLE_EQ(0.5, fourDecimalRound(result[1]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[2]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[3]));
EXPECT_DOUBLE_EQ(.0, fourDecimalRound(result[4]));
EXPECT_DOUBLE_EQ(0.3333, fourDecimalRound(result[5]));
EXPECT_DOUBLE_EQ(-0.25, fourDecimalRound(result[6]));
EXPECT_DOUBLE_EQ(-0.5, fourDecimalRound(result[7]));
EXPECT_DOUBLE_EQ(0.5, fourDecimalRound(result[8]));
init
2023-04-18 13:31:14 +08:00
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, std){
double *dataMA= new double [9]{1, 2, 3, 2, 2, 6, 3, 3, 6};
auto A = Aurora::Matrix::fromRawData(dataMA,3,3);
auto D= Aurora::std(A);
double* resultStd = D.getData();
EXPECT_DOUBLE_EQ(1.0, resultStd[0]);
EXPECT_DOUBLE_EQ(2.3094, fourDecimalRound(resultStd[1]));
EXPECT_DOUBLE_EQ(1.7321, fourDecimalRound(resultStd[2]));
init
2023-04-18 13:31:14 +08:00
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, fftAndComplexAndIfft){
Refactor Matrix unit test, remove log
2023-04-21 17:25:22 +08:00
// double input[10]{1,1,0,2,2,0,1,1,0,2};
// std::complex<double>* complexInput = Aurora::complex(10,input);
// //复数化后，实部不变，虚部全为0
// EXPECT_DOUBLE_EQ(complexInput[1].real(),1.0)<<" complex error";
// EXPECT_DOUBLE_EQ(complexInput[1].imag(),0)<<" complex error";
// std::complex<double>* result = Aurora::fft(10,complexInput);
// delete [] complexInput;
// //检验fft结果与matlab是否对应
// EXPECT_DOUBLE_EQ(0.0729, fourDecimalRound(result[1].real()))<<" fft result value error";
// EXPECT_DOUBLE_EQ(2.4899, fourDecimalRound(result[2].imag()))<<" fft result value error";
// //检验fft的结果是否共轭
// EXPECT_DOUBLE_EQ(0, result[4].imag()+result[6].imag())<<" fft result conjugate error";
// EXPECT_DOUBLE_EQ(0, result[4].real()-result[6].real())<<" fft result conjugate error";
// std::complex<double>* ifftResult = Aurora::ifft(10,result);
// EXPECT_DOUBLE_EQ(fourDecimalRound(ifftResult[1].real()),1.0)<<" ifft result real value error";
// EXPECT_DOUBLE_EQ(fourDecimalRound(ifftResult[1].imag()),0)<<" ifft result imag value error";
// delete [] result;
// delete [] ifftResult;
init
2023-04-18 13:31:14 +08:00
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, hilbert) {
init
2023-04-18 13:31:14 +08:00
double input[10]{1,1,0,2,2,0,1,1,0,2};
auto result = Aurora::hilbert(10,input);
EXPECT_DOUBLE_EQ(fourDecimalRound(result[1].real()),1.0)<<" hilbert result real value error";
EXPECT_DOUBLE_EQ(fourDecimalRound(result[1].imag()),0.3249)<<" hilbert result imag value error";
delete [] result;
result = Aurora::hilbert(9,input);
EXPECT_DOUBLE_EQ(fourDecimalRound(result[1].real()),1.0)<<" hilbert result real value error";
EXPECT_DOUBLE_EQ(fourDecimalRound(result[1].imag()),0.4253)<<" hilbert result imag value error";
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, interp2) {
Add repmat and interp unittest.
2023-04-21 17:12:53 +08:00
double* xD= new double[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{4});
double* yD= new double[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<double>(yD,std::default_delete<double[]>()),std::vector<int>{5});
double* x1D= new double[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<double>(x1D,std::default_delete<double[]>()),std::vector<int>{3});
double* y1D= new double[3]{4.5823,10.552,18};
Aurora::Matrix y1(std::shared_ptr<double>(y1D,std::default_delete<double[]>()),std::vector<int>{3});
double* vD= new double[20]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v(std::shared_ptr<double>(vD,std::default_delete<double[]>()),std::vector<int>{5,4});
Aurora::Matrix result = interp2(x,y,v,x1,y1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],303.5663);
EXPECT_DOUBLE_AE(result.getData()[1],243.6711);
EXPECT_DOUBLE_AE(result.getData()[2],-47.32);
result = interp2(x,y,v,x1,y1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],302.6298);
EXPECT_DOUBLE_AE(result.getData()[1],261.1313);
EXPECT_DOUBLE_AE(result.getData()[2],18.6089);
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, interp3) {
Add repmat and interp unittest.
2023-04-21 17:12:53 +08:00
double* xD= new double[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{4});
double* yD= new double[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<double>(yD,std::default_delete<double[]>()),std::vector<int>{5});
double* zD= new double[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<double>(zD,std::default_delete<double[]>()),std::vector<int>{3});
double* x1D= new double[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<double>(x1D,std::default_delete<double[]>()),std::vector<int>{3});
double* y1D= new double[3]{4.5823,10.552,18};
Aurora::Matrix y1(std::shared_ptr<double>(y1D,std::default_delete<double[]>()),std::vector<int>{3});
double* z1D= new double[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<double>(z1D,std::default_delete<double[]>()),std::vector<int>{3});
double* vD= new double[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v(std::shared_ptr<double>(vD,std::default_delete<double[]>()),std::vector<int>{5,4,3});
Aurora::Matrix result = interp3(x,y,z,v,x1,y1,z1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],303.5663);
EXPECT_DOUBLE_AE(result.getData()[1],243.6711);
EXPECT_DOUBLE_AE(result.getData()[2],-47.32);
result = interp3(x,y,z,v,x1,y1,z1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],302.6298);
EXPECT_DOUBLE_AE(result.getData()[1],261.1313);
EXPECT_DOUBLE_AE(result.getData()[2],18.6089);
}
Arrange Test directory.
2023-04-23 09:30:47 +08:00
TEST_F(Function2D_Test, interpn) {
Add repmat and interp unittest.
2023-04-21 17:12:53 +08:00
double* xD= new double[4]{1,2,3,4};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{4});
double* yD= new double[5]{1,4,9,16,25};
Aurora::Matrix y(std::shared_ptr<double>(yD,std::default_delete<double[]>()),std::vector<int>{5});
double* zD= new double[3]{22,33,56};
Aurora::Matrix z(std::shared_ptr<double>(zD,std::default_delete<double[]>()),std::vector<int>{3});
double* x1D= new double[3]{1.22,2.69,3.84};
Aurora::Matrix x1(std::shared_ptr<double>(x1D,std::default_delete<double[]>()),std::vector<int>{3});
double* y1D= new double[3]{2.5823,3.552,3.888};
Aurora::Matrix y1(std::shared_ptr<double>(y1D,std::default_delete<double[]>()),std::vector<int>{3});
double* z1D= new double[3]{22.256,40.22,48};
Aurora::Matrix z1(std::shared_ptr<double>(z1D,std::default_delete<double[]>()),std::vector<int>{3});
double* v2D= new double[20]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v2(std::shared_ptr<double>(v2D,std::default_delete<double[]>()),std::vector<int>{5,4});
double* v3D= new double[60]{158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25,
158,256,332,428,22,56,487,54,26,333,541,554,445,48,22,54,36,98,11,25};
Aurora::Matrix v3(std::shared_ptr<double>(v3D,std::default_delete<double[]>()),std::vector<int>{5,4,3});
Aurora::Matrix result = interpn(y,x,v2,x1,y1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],399.2742);
EXPECT_DOUBLE_AE(result.getData()[1],387.7651);
EXPECT_DOUBLE_AE(result.getData()[2],135.6138);
result = interpn(y,x,v2,x1,y1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],352.1727);
EXPECT_DOUBLE_AE(result.getData()[1],269.8596);
EXPECT_DOUBLE_AE(result.getData()[2],94.7908);
result = interpn(y,x,z,v3,x1,y1,z1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_AE(result.getData()[0],399.2742);
EXPECT_DOUBLE_AE(result.getData()[1],387.7651);
EXPECT_DOUBLE_AE(result.getData()[2],135.6138);
result = interpn(y,x,z,v3,x1,y1,z1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_AE(result.getData()[0],352.1727);
EXPECT_DOUBLE_AE(result.getData()[1],269.8596);
EXPECT_DOUBLE_AE(result.getData()[2],94.7908);
}
Reference in New Issue Copy Permalink