DevGrid_test.cpp

 
#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
#include <doctest/doctest.h>
 
#include "include/CommonRestartMetadata.hpp"
#include "include/DevGrid.hpp"
#include "include/DevGridIO.hpp"
#include "include/IStructure.hpp"
#include "include/NZLevels.hpp"
 
#include <cstdio>
#include <fstream>
 
namespace Nextsim {
 
const std::string stateFilename = "modelState.test.nc";
 
TEST_SUITE_BEGIN("DevGrid");
TEST_CASE("Write and read a ModelState-based DevGrid restart file")
{
    DevGrid grid;
    grid.setIO(new DevGridIO(grid));
 
    // Fill in the data. It is not real data.
    size_t nx = DevGrid::nx;
    size_t ny = DevGrid::nx;
    double yFactor = 0.01;
    double xFactor = 0.0001;
 
    NZLevels::set(1);
    ModelArray::setDimensions(ModelArray::Type::H, { nx, ny });
    ModelArray::setDimensions(ModelArray::Type::Z, { nx, ny, NZLevels::get() });
 
    HField fractional(ModelArray::Type::H);
    HField mask(ModelArray::Type::H);
    for (int j = 0; j < ny; ++j) {
        for (int i = 0; i < nx; ++i) {
            fractional(i, j) = j * yFactor + i * xFactor;
            mask(i, j) = (i + j) % 2;
        }
    }
 
    HField hice = fractional + 1;
    HField cice = fractional + 2;
    HField sst = fractional + 3;
    HField sss = fractional + 4;
    HField hsnow = fractional + 5;
 
    HField ticeValue = -(fractional + 1);
    ZField tice = ModelArray::ZField();
    tice.setData(ticeValue);
 
    ModelState state = {{
        { "mask", mask },
        { "hice", hice },
        { "cice", cice },
        { "sst", sst },
        { "sss", sss },
        { "hsnow", hsnow },
        { "tice", tice },
    }, {}};
 
    ModelMetadata metadata;
    metadata.setTime(TimePoint("2000-01-01T00:00:00Z"));
 
    grid.dumpModelState(state, metadata, stateFilename);
 
    ModelArray::setDimensions(ModelArray::Type::H, { 1, 1 });
    REQUIRE(ModelArray::dimensions(ModelArray::Type::H)[0] == 1);
    DevGrid gridIn;
    size_t targetX = 3;
    size_t targetY = 7;
 
    gridIn.setIO(new DevGridIO(grid));
    ModelState ms = gridIn.getModelState(stateFilename);
 
    REQUIRE(ModelArray::dimensions(ModelArray::Type::H)[0] == DevGrid::nx);
    REQUIRE(ModelArray::dimensions(ModelArray::Type::H)[1] == DevGrid::nx);
    REQUIRE(ms.data.at("hice")(targetX, targetY) != 0);
    REQUIRE(ms.data.at("hice")(targetX, targetY) > 1);
    REQUIRE(ms.data.at("hice")(targetX, targetY) < 2);
    REQUIRE(ms.data.at("hice")(targetX, targetY) == 1.0703);
 
    ZField ticeIn = ms.data.at("tice");
 
    REQUIRE(ticeIn.dimensions()[2] == 1);
    REQUIRE(ticeIn(targetX, targetY, 0U) == -1.0703);
 
    std::remove(stateFilename.c_str());
}
TEST_SUITE_END();
 
}