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

Add some compareSet functions

Browse Source
This commit is contained in:
kradchen
2023-06-09 14:32:02 +08:00
parent 4077843d13
commit 611da0872b
3 changed files with 96 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/Function1D_Test.cpp
View File

@@ -28,8 +28,10 @@ protected:
};
TEST_F(Function1D_Test,compareSet){
double * dataA =Aurora::malloc(9);
double * dataB =Aurora::malloc(9);
for (int i = 0; i < 9; ++i) {
dataA[i]=(double)(i-3);
dataB[i]=(double)(i+2);
}
Aurora::Matrix A = Aurora::Matrix::New(dataA,3,3);
EXPECT_EQ(-3, A[0]);
@@ -39,6 +41,22 @@ TEST_F(Function1D_Test,compareSet){
EXPECT_EQ(0, A[0]);
EXPECT_EQ(0, A[1]);
EXPECT_EQ(0, A[2]);
Aurora::Matrix B = Aurora::Matrix::New(dataB,3,3);
compareSet(A,B,4,-1,Aurora::NE);
EXPECT_EQ(-1, A[0]);
EXPECT_EQ(-1, A[1]);
EXPECT_EQ(0, A[2]);
EXPECT_EQ(-1, A[3]);
compareSet(A,B,9,Aurora::LT);
EXPECT_EQ(9, A[0]);
EXPECT_EQ(9, A[1]);
EXPECT_EQ(9, A[2]);
EXPECT_EQ(9, A[3]);
compareSet(A,9,B,Aurora::EQ);
EXPECT_EQ(2, A[0]);
EXPECT_EQ(3, A[1]);
EXPECT_EQ(4, A[2]);
EXPECT_EQ(5, A[3]);
}