MockStop.hpp

Go to the documentation of this file.

#pragma once
 
#include "models/strategies/stop/StopRule.hpp"
 
#include <nlohmann/json.hpp>
 
#include <set>
#include <string>
 
namespace ppforest2::test {
  class MockStop : public stop::StopRule {
  public:
    bool compute_return;
    mutable int compute_calls = 0;
 
    explicit MockStop(bool compute_return = false)
        : compute_return(compute_return) {}
 
    nlohmann::json to_json() const override { return {{"name", "mock_stop"}}; }
    std::string display_name() const override { return "Mock Stop"; }
    std::set<types::Mode> supported_modes() const override {
      return {types::Mode::Classification, types::Mode::Regression};
    }
 
  protected:
    bool compute(NodeContext const& /*ctx*/, stats::RNG& /*rng*/) const override {
      ++compute_calls;
      return compute_return;
    }
  };
}