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

Add prod, dot and their unit test.

Browse Source
This commit is contained in:
Krad
2023-04-25 17:28:33 +08:00
parent de1c1b0dda
commit cf96572074
3 changed files with 179 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
test/Function2D_Test.cpp
View File

@@ -342,6 +342,62 @@ TEST_F(Function2D_Test, hilbert) {
EXPECT_DOUBLE_EQ(fourDecimalRound(result[11].imag()),0.3249);
}
TEST_F(Function2D_Test, prod) {
double *dataB = new double[20]{1.1, 2.6, 3.8, 6.2,
4.3, 5.7, 6.9, 10.6,
7.1, 8.3, 9.7, 11.2,
17.8, 13.3,26.5 , -7.7,
9.9, 8.2, 6.3, 5.1};
auto B = Aurora::Matrix::fromRawData(dataB, 4, 5);
auto ret = Aurora::prod(B);
EXPECT_EQ(1, ret.getDimSize(0));
EXPECT_EQ(5, ret.getDimSize(1));
EXPECT_DOUBLE_EQ(0.0067, fourDecimalRound(ret.getData()[0]/10000));
EXPECT_DOUBLE_EQ(0.1793, fourDecimalRound(ret.getData()[1]/10000));
EXPECT_DOUBLE_EQ(0.6402, fourDecimalRound(ret.getData()[2]/10000));
EXPECT_DOUBLE_EQ(-4.8307, fourDecimalRound(ret.getData()[3]/10000));
EXPECT_DOUBLE_EQ(0.2608, fourDecimalRound(ret.getData()[4]/10000));
}
TEST_F(Function2D_Test, dot) {
{
double *dataA = new double[8];
double *dataB = new double[8];
for (int i = 0; i < 8; ++i) {
dataA[i] = (double) (i);
dataB[i] = (double) (2);
}
Aurora::Matrix A = Aurora::Matrix::fromRawData(dataA, 4, 2);
Aurora::Matrix B = Aurora::Matrix::fromRawData(dataB, 4, 2);
auto C = Aurora::dot(A, B);
EXPECT_EQ(1, C.getDimSize(0));
EXPECT_EQ(2, C.getDimSize(1));
EXPECT_DOUBLE_EQ(12, fourDecimalRound(C.getData()[0]));
EXPECT_DOUBLE_EQ(44, fourDecimalRound(C.getData()[1]));
}
{
double *dataB = new double[20]{1.1, 2.6, 3.8, 6.2,
4.3, 5.7, 6.9, 10.6,
7.1, 8.3, 9.7, 11.2,
17.8, 13.3,26.5 , -7.7,
9.9, 8.2, 6.3, 5.1};
auto B = Aurora::Matrix::fromRawData(dataB, 4, 5);
auto C = Aurora::dot(Aurora::fft(B),Aurora::fft(B));
EXPECT_EQ(1, C.getDimSize(0));
EXPECT_EQ(5, C.getDimSize(1));
EXPECT_DOUBLE_EQ(0.2434, fourDecimalRound(C.getData()[0]/1000));
EXPECT_DOUBLE_EQ(0., fourDecimalRound(C.getData()[1]/1000));
EXPECT_DOUBLE_EQ(0.8438, fourDecimalRound(C.getData()[2]/1000));
EXPECT_DOUBLE_EQ(0., fourDecimalRound(C.getData()[3]/1000));
EXPECT_DOUBLE_EQ(1.3553, fourDecimalRound(C.getData()[4]/1000));
EXPECT_DOUBLE_EQ(0., fourDecimalRound(C.getData()[5]/1000));
EXPECT_DOUBLE_EQ(5.0211, fourDecimalRound(C.getData()[6]/1000));
EXPECT_DOUBLE_EQ(0., fourDecimalRound(C.getData()[7]/1000));
EXPECT_DOUBLE_EQ(0.9238, fourDecimalRound(C.getData()[8]/1000));
EXPECT_DOUBLE_EQ(0., fourDecimalRound(C.getData()[9]/1000));
}
}
TEST_F(Function2D_Test, 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});