Add meshgrid and unitTest.

This commit is contained in:
sunwen
2023-06-08 15:26:13 +08:00
parent 36d8866667
commit fe84729646
3 changed files with 75 additions and 0 deletions

View File

@@ -9,6 +9,7 @@
#include "common/dataBlockCreation/getAscanBlock.h"
#include "common/dataBlockCreation/blockingGeometryInfo.h"
#include "common/dataBlockCreation/removeDataFromArrays.h"
#include "common/meshgrid.h"
#include "Parser.h"
#include "MatlabReader.h"
@@ -340,3 +341,32 @@ TEST_F(Common_Test, removeDataFromArrays) {
EXPECT_DOUBLE_EQ(result[2], 5);
EXPECT_DOUBLE_EQ(result[3], 6);
}
TEST_F(Common_Test, meshgrid) {
MatlabReader m("/home/sun/testData/meshgrid.mat");
auto x = m.read("x");
auto y = m.read("y");
auto z = m.read("z");
auto result = Recon::meshgrid(x, y, z);
auto xx = m.read("xx");
auto yy = m.read("yy");
auto zz = m.read("zz");
EXPECT_DOUBLE_AE(xx.getDataSize(), result.xx.getDataSize());
for(int i=0; i<xx.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(xx[i], result.xx[i]);
}
EXPECT_DOUBLE_AE(yy.getDataSize(), result.yy.getDataSize());
for(int i=0; i<yy.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(yy[i], result.yy[i]);
}
EXPECT_DOUBLE_AE(zz.getDataSize(), result.zz.getDataSize());
for(int i=0; i<zz.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(zz[i], result.zz[i]);
}
}