sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add repmat and interp unittest.

Browse Source
This commit is contained in:
sunwen
2023-04-21 17:12:53 +08:00
parent 8f7626d717
commit c1c09fa45f
1 changed files with 158 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

158
test/FunctionTester.cpp
View File

@@ -1,8 +1,10 @@
#include <gtest/gtest.h>
#include <vector>
#include "Matrix.h"
#include "Function.h"
#include "Function1D.h"
#include "Function2D.h"
#include "Function3D.h"
#define DISPLAY_MATRIX(Matrix)\
@@ -31,6 +33,8 @@ double fourDecimalRound(double src){
return round(src*10000.0)/10000.0;
}
#define EXPECT_DOUBLE_AE(valueA,valueB)\
EXPECT_DOUBLE_EQ(fourDecimalRound(valueA),fourDecimalRound(valueB));
TEST_F(FunctionTester, MatrixCreate) {
double * dataA =Aurora::malloc(9);
@@ -396,3 +400,157 @@ TEST_F(FunctionTester, hilbert) {
EXPECT_DOUBLE_EQ(fourDecimalRound(result[1].imag()),0.4253)<<" hilbert result imag value error";
}
TEST_F(FunctionTester, repmat) {
//input 1D repmat 2D
double* input = new double[10]{1,2,3,4,5,6,7,8,9,10};
Aurora::Matrix m(std::shared_ptr<double>(input,std::default_delete<double[]>()),std::vector<int>{10});
EXPECT_DOUBLE_EQ(repmat(m,1,1).getData()[7],8);
EXPECT_DOUBLE_EQ(repmat(m,2,3).getDataSize(),60);
EXPECT_DOUBLE_EQ(repmat(m,2,3).getData()[22],3);
EXPECT_DOUBLE_EQ(repmat(m,2,3).getDimSize(0),20);
EXPECT_DOUBLE_EQ(repmat(m,2,3).getDimSize(1),3);
EXPECT_DOUBLE_EQ(repmat(m,2,3).getDimSize(2),1);
//input 2D repmat 2D
double* input2 = new double[10]{1,2,3,4,5,6,7,8,9,10};
Aurora::Matrix m2(std::shared_ptr<double>(input2,std::default_delete<double[]>()),std::vector<int>{5,2});
EXPECT_DOUBLE_EQ(repmat(m2,1,1).getData()[7],8);
EXPECT_DOUBLE_EQ(repmat(m2,2,3).getDataSize(),60);
EXPECT_DOUBLE_EQ(repmat(m2,2,3).getData()[22],3);
EXPECT_DOUBLE_EQ(repmat(m2,2,3).getDimSize(0),10);
EXPECT_DOUBLE_EQ(repmat(m2,2,3).getDimSize(1),6);
EXPECT_DOUBLE_EQ(repmat(m2,2,3).getDimSize(2),1);
//input 1D repmat 3D
double* input3 = new double[10]{1,2,3,4,5,6,7,8,9,10};
Aurora::Matrix m3(std::shared_ptr<double>(input3,std::default_delete<double[]>()),std::vector<int>{10});
EXPECT_DOUBLE_EQ(repmat(m3,1,1,1).getData()[7],8);
EXPECT_DOUBLE_EQ(repmat(m3,2,3,4).getDataSize(),240);
EXPECT_DOUBLE_EQ(repmat(m3,2,3,4).getData()[22],3);
EXPECT_DOUBLE_EQ(repmat(m3,2,3,4).getDimSize(0),20);
EXPECT_DOUBLE_EQ(repmat(m3,2,3,4).getDimSize(1),3);
EXPECT_DOUBLE_EQ(repmat(m3,2,3,4).getDimSize(2),4);
//input 2D repmat 3D
double* input4 = new double[10]{1,2,3,4,5,6,7,8,9,10};
Aurora::Matrix m4(std::shared_ptr<double>(input4,std::default_delete<double[]>()),std::vector<int>{5,2});
EXPECT_DOUBLE_EQ(repmat(m4,1,1).getData()[7],8);
EXPECT_DOUBLE_EQ(repmat(m4,2,3,4).getDataSize(),240);
EXPECT_DOUBLE_EQ(repmat(m4,2,3,4).getData()[22],3);
EXPECT_DOUBLE_EQ(repmat(m4,2,3,4).getDimSize(0),10);
EXPECT_DOUBLE_EQ(repmat(m4,2,3,4).getDimSize(1),6);
EXPECT_DOUBLE_EQ(repmat(m4,2,3,4).getDimSize(2),4);
}
TEST_F(FunctionTester, interp1) {
double* xD= new double[5]{1,2,3,4,5};
Aurora::Matrix x(std::shared_ptr<double>(xD,std::default_delete<double[]>()),std::vector<int>{5});
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});
Aurora::Matrix result = interp1(x,y,x1,Aurora::InterpnMethod::Spline);
EXPECT_DOUBLE_EQ(result.getData()[0],1.4884);
EXPECT_DOUBLE_EQ(result.getData()[1],7.2361);
EXPECT_DOUBLE_EQ(result.getData()[2],14.7456);
result = interp1(x,y,x1,Aurora::InterpnMethod::Linear);
EXPECT_DOUBLE_EQ(result.getData()[0],1.66);
EXPECT_DOUBLE_EQ(result.getData()[1],7.45);
EXPECT_DOUBLE_EQ(result.getData()[2],14.88);
}
TEST_F(FunctionTester, interp2) {
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);
}
TEST_F(FunctionTester, interp3) {
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);
}
TEST_F(FunctionTester, interpn) {
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);
}