Add vertcat

This commit is contained in:
kradchen
2023-06-09 15:13:52 +08:00
parent 611da0872b
commit 428d75e548
3 changed files with 38 additions and 0 deletions

View File

@@ -431,6 +431,21 @@ TEST_F(Function1D_Test, horzcat) {
EXPECT_DOUBLE_EQ(result.getDimSize(1),2);
}
TEST_F(Function1D_Test, vertcat) {
double *data1 = new double[6]{1,2,3,4,5,6};
auto matrix1 = Aurora::Matrix::fromRawData(data1, 2,3);
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::vertcat(matrix1,matrix2);
EXPECT_DOUBLE_EQ(result.getData()[0],1);
EXPECT_DOUBLE_EQ(result.getData()[1],2);
EXPECT_DOUBLE_EQ(result.getData()[5],3);
EXPECT_DOUBLE_EQ(result.getData()[8],11);
EXPECT_DOUBLE_EQ(result.getDimSize(0),5);
EXPECT_DOUBLE_EQ(result.getDimSize(1),3);
}
TEST_F(Function1D_Test, vecnrom) {
//1Dim
double *data = new double[3]{1,2,-3};