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

Add Matrix method setBlockValue

Browse Source
This commit is contained in:
kradchen
2023-05-29 11:09:09 +08:00
parent 320d6c81cb
commit ce91b7a868
3 changed files with 94 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
test/Matrix_Test.cpp
View File

@@ -5,11 +5,9 @@
#include "Function1D.h"
#include "Function2D.h"
#include "Function3D.h"
#include "TestUtility.h"
#define DISPLAY_MATRIX
class Matrix_Test : public ::testing::Test {
protected:
static void SetUpMatrixTester() {
@@ -401,4 +399,32 @@ TEST_F(Matrix_Test, matrixfunction){
EXPECT_DOUBLE_EQ(C.getData()[6], 24);
EXPECT_DOUBLE_EQ(C.getData()[7], 306);
EXPECT_DOUBLE_EQ(C.getData()[8], 460);
double *dataE = Aurora::random(60);
Aurora::Matrix E = Aurora::Matrix::New(dataE, 3, 4, 5);
Aurora::Matrix block1 = E.block(0, 1, 2);
for (size_t i = 0; i < block1.getDimSize(2); i++)
{
for (size_t j = 0; j < block1.getDimSize(1); j++)
{
for (size_t k = 1; k < E.getDimSize(0); k++)
{
auto index1 = k+j*E.getDimSize(0)+i*E.getDimSize(1)*E.getDimSize(0);
auto index2 = k-1+j*block1.getDimSize(0)+i*block1.getDimSize(1)*block1.getDimSize(0);
EXPECT_DOUBLE_AE(E[index1],block1[index2]);
}
}
}
E.setBlockValue(0, 1, 2,-1);
for (size_t i = 0; i < block1.getDimSize(2); i++)
{
for (size_t j = 0; j < block1.getDimSize(1); j++)
{
for (size_t k = 1; k < E.getDimSize(0); k++)
{
auto index1 = k+j*E.getDimSize(0)+i*E.getDimSize(1)*E.getDimSize(0);
EXPECT_DOUBLE_AE(E[index1],-1);
}
}
}
}