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

Add isfinite and isnan and their test.

Browse Source
This commit is contained in:
kradchen
2023-05-16 10:24:09 +08:00
parent b2bacd8894
commit 1a8607eece
3 changed files with 44 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
test/Function1D_Test.cpp
View File

@@ -1,5 +1,6 @@
#include <cstddef>
#include <gtest/gtest.h>
#include <limits>
#include <vector>
#include "MatlabReader.h"
@@ -522,5 +523,24 @@ TEST_F(Function1D_Test, convertfp16tofloat) {
for (size_t i = 0; i<count; i++) {
EXPECT_EQ(result[i], output.getData()[i])<<"index:"<<i<<",input:"<<input.get()[i];
}
}
TEST_F(Function1D_Test, nanAndFinite){
double n=std::nan("");
double infinite=std::numeric_limits<double>::infinity();
double* data = new double[9]{infinite,3,n,2,2,1,4,4,7};
auto matrix = Aurora::Matrix::fromRawData(data, 9,1,1);
auto r = Aurora::isnan(matrix);
EXPECT_DOUBLE_EQ(r.getData()[0],0);
EXPECT_DOUBLE_EQ(r.getData()[1],0);
EXPECT_DOUBLE_EQ(r.getData()[2],1);
r = Aurora::isfinite(matrix);
EXPECT_DOUBLE_EQ(r.getData()[0],0);
EXPECT_DOUBLE_EQ(r.getData()[1],1);
EXPECT_DOUBLE_EQ(r.getData()[2],0);
Aurora::nantoval(matrix,1);
EXPECT_DOUBLE_EQ(matrix.getData()[2],1);
}