windowed_adaptation_test.cpp

#include <stan/mcmc/windowed_adaptation.hpp>
#include <test/unit/services/instrumented_callbacks.hpp>
#include <gtest/gtest.h>

TEST(McmcWindowedAdaptation, set_window_params1) {
  stan::test::unit::instrumented_logger logger;

  stan::mcmc::windowed_adaptation adapter("test");

  adapter.set_window_params(10, 1, 1, 1, logger);
  ASSERT_EQ(logger.call_count(), logger.call_count_info());
  EXPECT_EQ(1, logger.find_info("WARNING: No test estimation is"));
  EXPECT_EQ(1, logger.find_info("performed for num_warmup < 20"));
}

TEST(McmcWindowedAdaptation, set_window_params2) {
  stan::test::unit::instrumented_logger logger;

  stan::mcmc::windowed_adaptation adapter("test");

  adapter.set_window_params(100, 75, 50, 25, logger);
  ASSERT_EQ(logger.call_count(), logger.call_count_info());
  EXPECT_EQ(1,
            logger.find_info(
                "WARNING: There aren't enough warmup iterations to fit the"));
  EXPECT_EQ(
      1, logger.find_info(
             "         three stages of adaptation as currently configured."));
  EXPECT_EQ(1,
            logger.find_info(
                "         Reducing each adaptation stage to 15%/75%/10% of"));
  EXPECT_EQ(
      1, logger.find_info("         the given number of warmup iterations:"));
  EXPECT_EQ(1, logger.find_info("           init_buffer = 15"));
  EXPECT_EQ(1, logger.find_info("           adapt_window = 75"));
  EXPECT_EQ(1, logger.find_info("           term_buffer = 10"));
}

TEST(McmcWindowedAdaptation, set_window_params3) {
  stan::test::unit::instrumented_logger logger;

  stan::mcmc::windowed_adaptation adapter("test");

  adapter.set_window_params(1000, 75, 50, 25, logger);

  ASSERT_EQ(0, logger.call_count());
  ASSERT_EQ(0, logger.call_count_info());
}

TEST(McmcWindowedAdaptation, finished) {
  stan::test::unit::instrumented_logger logger;

  stan::mcmc::windowed_adaptation adapter("test");

  adapter.set_window_params(1000, 75, 50, 25, logger);

  for (size_t i = 0; i < 999; i++) {
    EXPECT_FALSE(adapter.finished());
    adapter.increment_window_counter();
  }
  EXPECT_TRUE(adapter.finished());
}