NodeContextFixture.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/strategies/NodeContext.hpp"
#include "stats/GroupPartition.hpp"
#include "stats/Stats.hpp"
#include "utils/Types.hpp"
 
#include <utility>
 
namespace ppforest2::test {
  namespace {
    inline types::GroupIdVector median_split_ids(int n) {
      types::GroupIdVector ids(n);
      int const mid = n / 2;
      for (int i = 0; i < n; ++i) {
        ids(i) = (i < mid) ? 0 : 1;
      }
      return ids;
    }
  }

  struct NodeContextFixture {
    types::FeatureMatrix x;
    types::GroupIdVector y_ids;
    stats::GroupPartition y_part;
    types::OutcomeVector y_vec;
    stats::RNG rng;
    NodeContext ctx;
 
    NodeContextFixture(types::FeatureMatrix x_in, types::GroupIdVector ids_in)
        : x(std::move(x_in))
        , y_ids(std::move(ids_in))
        , y_part(y_ids)
        , y_vec(y_ids.cast<types::Outcome>())
        , rng(0)
        , ctx(x, y_part, y_vec, 0) {}
 
    NodeContextFixture(types::FeatureMatrix x_in, types::OutcomeVector y_vec_in)
        : x(std::move(x_in))
        , y_ids(median_split_ids(static_cast<int>(y_vec_in.size())))
        , y_part(y_ids)
        , y_vec(std::move(y_vec_in))
        , rng(0)
        , ctx(x, y_part, y_vec, 0) {}
 
    NodeContextFixture(types::FeatureMatrix x_in, stats::GroupPartition y_part_in, types::OutcomeVector y_vec_in)
        : x(std::move(x_in))
        , y_ids()
        , y_part(std::move(y_part_in))
        , y_vec(std::move(y_vec_in))
        , rng(0)
        , ctx(x, y_part, y_vec, 0) {}
 
    NodeContextFixture(NodeContextFixture const&)            = delete;
    NodeContextFixture(NodeContextFixture&&)                 = delete;
    NodeContextFixture& operator=(NodeContextFixture const&) = delete;
    NodeContextFixture& operator=(NodeContextFixture&&)      = delete;
  };
}