test_determinist.cpp

// Copyright (c) 2023, INRAE.
// Distributed under the terms of the GPL-3 Licence.
// The full licence is in the file LICENCE, distributed with this software.

#include <fstream>
#include <vector>
#include <tuple>
#include <array>

#include <gtest/gtest.h>

#include <xtensor/xtensor.hpp>
#include <xtensor/xview.hpp>
#include <xtensor/xmanipulation.hpp>
#include <xtensor/xcsv.hpp>

#include "evalhyd/evald.hpp"

#ifndef EVALHYD_DATA_DIR
#error "need to define data directory"
#endif

using namespace xt::placeholders;  // required for `_` to work


std::vector<std::string> all_metrics_d = {
        "RMSE", "NSE", "KGE", "KGEPRIME"
};

std::tuple<xt::xtensor<double, 2>, xt::xtensor<double, 2>> load_data_d()
{
    // read in data
    std::ifstream ifs;
    ifs.open(EVALHYD_DATA_DIR "/q_obs.csv");
    xt::xtensor<double, 2> observed = xt::load_csv<int>(ifs);
    ifs.close();

    ifs.open(EVALHYD_DATA_DIR "/q_prd.csv");
    xt::xtensor<double, 2> predicted = xt::view(
            xt::load_csv<double>(ifs), xt::range(0, 5), xt::all()
    );
    ifs.close();

    return std::make_tuple(observed, predicted);
}

TEST(DeterministTests, TestMetrics)
{
    // read in data
    xt::xtensor<double, 2> observed;
    xt::xtensor<double, 2> predicted;
    std::tie(observed, predicted) = load_data_d();

    // compute scores (with 2D tensors)
    std::vector<xt::xarray<double>> metrics =
            evalhyd::evald(
                    observed, predicted, {"RMSE", "NSE", "KGE", "KGEPRIME"}
            );

    // check results on all metrics
    xt::xtensor<double, 3> rmse =
            {{{777.034272}},
             {{776.878479}},
             {{777.800217}},
             {{778.151082}},
             {{778.61487 }}};
    EXPECT_TRUE(xt::allclose(metrics[0], rmse));

    xt::xtensor<double, 3> nse =
            {{{0.718912}},