Update discretizePositions unittest.

This commit is contained in:
sunwen
2023-05-31 13:53:26 +08:00
parent 44384649d4
commit 24bb49c46c

View File

@@ -1,9 +1,11 @@
#include <cstddef>
#include <gtest/gtest.h>
#include <limits>
#include "Function1D.h"
#include "MatlabReader.h"
#include "Matrix.h"
#include "config/config.h"
#include "transmissionReconstruction/reconstruction/buildMatrix/DGradient.h"
#include "transmissionReconstruction/reconstruction/buildMatrix/FMM.h"
#include "transmissionReconstruction/reconstruction/buildMatrix/buildMatrix.h"
@@ -66,12 +68,45 @@ TEST_F(Reconstruction_Test, getDimensions) {
}
TEST_F(Reconstruction_Test, discretizePositions) {
auto senderList = Aurora::Matrix::fromRawData(new double[6]{1, 2, 3, 1, 2, 4}, 3, 2);
auto receiverList = Aurora::Matrix::fromRawData(new double[6]{1, 8, 3, 1, 2, 1}, 3, 2);
double numPixelXY = 128;
auto result = Recon::discretizePositions(senderList,receiverList,numPixelXY);
EXPECT_DOUBLE_AE(128,result.receiverCoordList[1]);
EXPECT_DOUBLE_AE(1,result.senderCoordList[1]);
MatlabReader m("/home/sun/testData/discretizePositions.mat");
auto senderList = m.read("senderList");
auto receiverList = m.read("receiverList");
auto result = Recon::discretizePositions(senderList,receiverList,Recon::transParams::numPixelXY);
auto senderListResult = m.read("senderListResult");
auto receiverListResult = m.read("receiverListResult");
auto dims = m.read("dims");
auto ddims = m.read("ddims");
auto res = m.read("res");
EXPECT_DOUBLE_AE(senderListResult.getDataSize(), result.senderCoordList.getDataSize());
for(size_t i=0; i<senderListResult.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(senderListResult[i], result.senderCoordList[i]);
}
EXPECT_DOUBLE_AE(receiverListResult.getDataSize(), result.receiverCoordList.getDataSize());
for(size_t i=0; i<receiverListResult.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(receiverListResult[i], result.receiverCoordList[i]);
}
EXPECT_DOUBLE_AE(dims.getDataSize(), result.dims.getDataSize());
for(size_t i=0; i<dims.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(dims[i], result.dims[i]);
}
EXPECT_DOUBLE_AE(ddims.getDataSize(), result.ddims.getDataSize());
for(size_t i=0; i<ddims.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(ddims[i], result.ddims[i]);
}
EXPECT_DOUBLE_AE(res.getDataSize(), result.res.getDataSize());
for(size_t i=0; i<res.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(res[i], result.res[i]);
}
}
TEST_F(Reconstruction_Test, DGradient) {