#include "test_harness.h"
#include <TestProject/RNG/model/CompositeModel.h>
#include <TestProject/RNG/model/ConstantModel.h>
#include <TestProject/RNG/model/LinearModel.h>
using json = nlohmann::json;

TEST(composite_sums_base_and_noise) {
  auto base = std::make_unique<ConstantModel>(json::object(), 50.0f);
  auto noise = std::make_unique<LinearModel>(json{{"k", 2.0f}}, 0.0f);
  CompositeModel m(std::move(base), std::move(noise));
  CHECK_FLOAT_EQ(m.evaluate(0), 50.0f, 0.001f);
  CHECK_FLOAT_EQ(m.evaluate(5), 60.0f, 0.001f);
}

TEST(composite_both_constant) {
  auto base = std::make_unique<ConstantModel>(json::object(), 10.0f);
  auto noise = std::make_unique<ConstantModel>(json::object(), 3.0f);
  CompositeModel m(std::move(base), std::move(noise));
  CHECK_FLOAT_EQ(m.evaluate(0), 13.0f, 0.001f);
  CHECK_FLOAT_EQ(m.evaluate(999), 13.0f, 0.001f);
}

TEST(composite_evaluateBool_returns_false) {
  auto base = std::make_unique<ConstantModel>(json::object(), 1.0f);
  auto noise = std::make_unique<ConstantModel>(json::object(), 2.0f);
  CompositeModel m(std::move(base), std::move(noise));
  CHECK_EQ(m.evaluateBool(0), false);
}