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

Add horzcat and horzcat's unittest.

Browse Source
This commit is contained in:
sunwen
2023-04-25 15:17:15 +08:00
parent 493b84e4d1
commit de1c1b0dda
3 changed files with 47 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
test/Function1D_Test.cpp
View File

@@ -342,3 +342,29 @@ TEST_F(Function1D_Test, transpose) {
EXPECT_DOUBLE_EQ(result.getDimSize(0),2);
EXPECT_DOUBLE_EQ(result.getDimSize(1),3);
}
TEST_F(Function1D_Test, horzcat) {
double *data1 = new double[6]{1,2,3,4,5,6};
auto matrix1 = Aurora::Matrix::fromRawData(data1, 3,2);
double *data2 = new double[9]{7,8,9,10,11,12,13,14,15};
auto matrix2 = Aurora::Matrix::fromRawData(data2, 3,3);
auto result = Aurora::horzcat(matrix1,matrix2);
EXPECT_DOUBLE_EQ(result.getData()[0],1);
EXPECT_DOUBLE_EQ(result.getData()[1],2);
EXPECT_DOUBLE_EQ(result.getData()[10],11);
EXPECT_DOUBLE_EQ(result.getData()[14],15);
EXPECT_DOUBLE_EQ(result.getDimSize(0),3);
EXPECT_DOUBLE_EQ(result.getDimSize(1),5);
data1 = new double[6]{1,2,3,4,5,6};
matrix1 = Aurora::Matrix::fromRawData(data1, 3,1,1,Aurora::Complex);
data2 = new double[6]{7,8,9,10,11,12};
matrix2 = Aurora::Matrix::fromRawData(data2, 3,1,1,Aurora::Complex);
result = Aurora::horzcat(matrix1,matrix2);
EXPECT_DOUBLE_EQ(result.getData()[0],1);
EXPECT_DOUBLE_EQ(result.getData()[1],2);
EXPECT_DOUBLE_EQ(result.getData()[8],5);
EXPECT_DOUBLE_EQ(result.getData()[9],6);
EXPECT_DOUBLE_EQ(result.getDimSize(0),3);
EXPECT_DOUBLE_EQ(result.getDimSize(1),2);
}