From c1c09fa45fb2269ea9fc73c768085bfa2e8b41e8 Mon Sep 17 00:00:00 2001 From: sunwen Date: Fri, 21 Apr 2023 17:12:53 +0800 Subject: [PATCH] Add repmat and interp unittest. --- test/FunctionTester.cpp | 158 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 158 insertions(+) diff --git a/test/FunctionTester.cpp b/test/FunctionTester.cpp index ed86582..12c081f 100644 --- a/test/FunctionTester.cpp +++ b/test/FunctionTester.cpp @@ -1,8 +1,10 @@ #include +#include #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(input,std::default_delete()),std::vector{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(input2,std::default_delete()),std::vector{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(input3,std::default_delete()),std::vector{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(input4,std::default_delete()),std::vector{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(xD,std::default_delete()),std::vector{5}); + double* yD= new double[5]{1,4,9,16,25}; + Aurora::Matrix y(std::shared_ptr(yD,std::default_delete()),std::vector{5}); + double* x1D= new double[3]{1.22,2.69,3.84}; + Aurora::Matrix x1(std::shared_ptr(x1D,std::default_delete()),std::vector{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(xD,std::default_delete()),std::vector{4}); + double* yD= new double[5]{1,4,9,16,25}; + Aurora::Matrix y(std::shared_ptr(yD,std::default_delete()),std::vector{5}); + double* x1D= new double[3]{1.22,2.69,3.84}; + Aurora::Matrix x1(std::shared_ptr(x1D,std::default_delete()),std::vector{3}); + double* y1D= new double[3]{4.5823,10.552,18}; + Aurora::Matrix y1(std::shared_ptr(y1D,std::default_delete()),std::vector{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(vD,std::default_delete()),std::vector{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(xD,std::default_delete()),std::vector{4}); + double* yD= new double[5]{1,4,9,16,25}; + Aurora::Matrix y(std::shared_ptr(yD,std::default_delete()),std::vector{5}); + double* zD= new double[3]{22,33,56}; + Aurora::Matrix z(std::shared_ptr(zD,std::default_delete()),std::vector{3}); + double* x1D= new double[3]{1.22,2.69,3.84}; + Aurora::Matrix x1(std::shared_ptr(x1D,std::default_delete()),std::vector{3}); + double* y1D= new double[3]{4.5823,10.552,18}; + Aurora::Matrix y1(std::shared_ptr(y1D,std::default_delete()),std::vector{3}); + double* z1D= new double[3]{22.256,40.22,48}; + Aurora::Matrix z1(std::shared_ptr(z1D,std::default_delete()),std::vector{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(vD,std::default_delete()),std::vector{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(xD,std::default_delete()),std::vector{4}); + double* yD= new double[5]{1,4,9,16,25}; + Aurora::Matrix y(std::shared_ptr(yD,std::default_delete()),std::vector{5}); + double* zD= new double[3]{22,33,56}; + Aurora::Matrix z(std::shared_ptr(zD,std::default_delete()),std::vector{3}); + double* x1D= new double[3]{1.22,2.69,3.84}; + Aurora::Matrix x1(std::shared_ptr(x1D,std::default_delete()),std::vector{3}); + double* y1D= new double[3]{2.5823,3.552,3.888}; + Aurora::Matrix y1(std::shared_ptr(y1D,std::default_delete()),std::vector{3}); + double* z1D= new double[3]{22.256,40.22,48}; + Aurora::Matrix z1(std::shared_ptr(z1D,std::default_delete()),std::vector{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(v2D,std::default_delete()),std::vector{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(v3D,std::default_delete()),std::vector{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); +} +