Add union and union's unittest.

This commit is contained in:
sunwen
2023-04-26 15:57:29 +08:00
parent b4423b756e
commit 9a3ec2805e
3 changed files with 40 additions and 0 deletions

View File

@@ -439,3 +439,19 @@ TEST_F(Function1D_Test, linspace) {
EXPECT_DOUBLE_AE(result.getData()[5], 3.3333);
EXPECT_DOUBLE_AE(result.getData()[6], 5.0000);
}
TEST_F(Function1D_Test, auroraUnion) {
double* data1 = new double[9]{3,3,2,2,2,1,4,4,7};
auto matrix1 = Aurora::Matrix::fromRawData(data1, 9,1,1);
double* data2 = new double[8]{6,6,7,7,8,1,2};
auto matrix2 = Aurora::Matrix::fromRawData(data2, 7,1,1);
auto result = Aurora::auroraUnion(matrix1, matrix2);
EXPECT_DOUBLE_AE(result.getData()[0],1);
EXPECT_DOUBLE_AE(result.getData()[1],2);
EXPECT_DOUBLE_AE(result.getData()[2],3);
EXPECT_DOUBLE_AE(result.getData()[3],4);
EXPECT_DOUBLE_AE(result.getData()[4],6);
EXPECT_DOUBLE_AE(result.getData()[5],7);
EXPECT_DOUBLE_AE(result.getData()[6],8);
}