Add compareSet to 1D

2023-06-08 15:59:57 +08:00
parent a0db3777d2
commit 1a29baeef8
3 changed files with 43 additions and 0 deletions
--- a/test/Function1D_Test.cpp
+++ b/test/Function1D_Test.cpp
@@ -26,6 +26,21 @@ protected:
    }

 };
+TEST_F(Function1D_Test,compareSet){
+    double * dataA =Aurora::malloc(9);
+    for (int i = 0; i < 9; ++i) {
+        dataA[i]=(double)(i-3);
+    }
+    Aurora::Matrix A = Aurora::Matrix::New(dataA,3,3);
+    EXPECT_EQ(-3, A[0]);
+    EXPECT_EQ(-2, A[1]);
+    EXPECT_EQ(-1, A[2]);
+    compareSet(A,0,0,Aurora::LT);
+    EXPECT_EQ(0, A[0]);
+    EXPECT_EQ(0, A[1]);
+    EXPECT_EQ(0, A[2]);
+}
+

 TEST_F(Function1D_Test, sign) {
    double * dataA =Aurora::malloc(9);