geopro/tests/data/test_local_repo.cpp

29 lines
976 B
C++

#include <gtest/gtest.h>
#include "repo/LocalSampleRepository.hpp"
using namespace geopro::data;
using namespace geopro::core;
static const std::string kDir = "D:/Git/lanbingtech/geopro/docs/剖面网格数据的色阶数据2等文件/";
TEST(LocalRepo, StructureNonEmpty) {
LocalSampleRepository repo(kDir);
auto gs = repo.loadStructure();
ASSERT_FALSE(gs.empty());
ASSERT_FALSE(gs[0].tms.empty());
ASSERT_FALSE(gs[0].tms[0].dss.empty());
EXPECT_EQ(gs[0].tms[0].dss[0].id, "grid1");
}
TEST(LocalRepo, LoadGridScatterAnomaly) {
LocalSampleRepository repo(kDir);
Grid g = repo.loadGrid("grid1");
EXPECT_EQ(g.nx(), 100); EXPECT_EQ(g.ny(), 22);
ScatterField s = repo.loadScatter("grid1");
EXPECT_EQ(s.v.size(), 2597u);
auto an = repo.loadAnomalies("grid1");
EXPECT_EQ(an.size(), 3u);
EXPECT_EQ(an[0].markType, AnomalyMarkType::Polyline);
ColorScale cs = repo.loadColorScale("grid1");
EXPECT_FALSE(cs.empty());
}