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

Add sub2ind and test

Browse Source
This commit is contained in:
kradchen
2023-05-10 13:46:40 +08:00
parent 61baf0dc4a
commit cbbcca5ceb
4 changed files with 59 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/Function1D_Test.cpp
View File

@@ -118,7 +118,7 @@ TEST_F(Function1D_Test, polyval){
auto matrixP = Aurora::Matrix::fromRawData(dataP,6);
auto matrixX = Aurora::Matrix::fromRawData(dataX,1);
auto result = Aurora::polyval(matrixP,matrixX);
EXPECT_DOUBLE_EQ(1495.6, result.getData()[0]);
EXPECT_DOUBLE_EQ(1495.5717310876246, result.getData()[0]);
// EXPECT_DOUBLE_EQ(162., result.getData()[1]);
// EXPECT_DOUBLE_EQ(262., result.getData()[2]);
}

17
test/Function2D_Test.cpp
View File

@@ -480,3 +480,20 @@ TEST_F(Function2D_Test, interp2) {
EXPECT_DOUBLE_AE(result.getData()[2],18.6089);
}
TEST_F(Function2D_Test, sub2ind) {
double* dI1= new double[4]{1,2,1,2};
Aurora::Matrix I1(std::shared_ptr<double>(dI1,std::default_delete<double[]>()),std::vector<int>{4});
double* dI2= new double[4]{2,2,1,1};
Aurora::Matrix I2(std::shared_ptr<double>(dI2,std::default_delete<double[]>()),std::vector<int>{4});
double* dI4= new double[4]{1,1,2,2};
Aurora::Matrix I3(std::shared_ptr<double>(dI4,std::default_delete<double[]>()),std::vector<int>{4});
double* dsz= new double[3]{2,2,2};
Aurora::Matrix sz(std::shared_ptr<double>(dsz,std::default_delete<double[]>()),std::vector<int>{3});
auto ma = Aurora::sub2ind(sz, {I1, I2, I3});
EXPECT_DOUBLE_EQ(3, ma.getData()[0]);
EXPECT_DOUBLE_EQ(4, ma.getData()[1]);
EXPECT_DOUBLE_EQ(5, ma.getData()[2]);
EXPECT_DOUBLE_EQ(6, ma.getData()[3]);
}