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

Add zeros and ones to Function3D.

Browse Source
This commit is contained in:
Krad
2023-04-23 13:29:53 +08:00
parent f4d59813cf
commit 3833040672
4 changed files with 122 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
test/Function3D_Test.cpp
View File

@@ -91,4 +91,52 @@ TEST_F(Function3D_Test, interpn) {
EXPECT_DOUBLE_AE(result.getData()[0],352.1727);
EXPECT_DOUBLE_AE(result.getData()[1],269.8596);
EXPECT_DOUBLE_AE(result.getData()[2],94.7908);
}
TEST_F(Function3D_Test, zerosAndones){
Aurora::Matrix zerosM = Aurora::zeros( 3, 4,5);
EXPECT_EQ(60,zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
}
zerosM = Aurora::zeros( 3, 4);
EXPECT_EQ(12,zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
}
zerosM = Aurora::zeros( 3, 1);
EXPECT_EQ(3,zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
}
zerosM = Aurora::zeros( 20);
EXPECT_EQ(400,zerosM.getDataSize());
for (int i = 0; i < zerosM.getDataSize(); ++i) {
EXPECT_EQ(0,zerosM.getData()[i])<<" error at index："<<i;
}
Aurora::Matrix onesM = Aurora::ones( 9, 9,9);
EXPECT_EQ(729,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9, 9);
EXPECT_EQ(81,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9, 1);
EXPECT_EQ(9,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
onesM = Aurora::ones( 9);
EXPECT_EQ(81,onesM.getDataSize());
for (int i = 0; i < onesM.getDataSize(); ++i) {
EXPECT_DOUBLE_EQ(1.0,onesM.getData()[i])<<" error at index："<<i;
}
}