#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());
}