test_on_effective_size_drop.cpp

Go to the documentation of this file.

// Copyright 2024 Ekumen, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#include <gtest/gtest.h>
 
#include <vector>
 
#include "beluga/policies/on_effective_size_drop.hpp"
 
namespace {
 
TEST(OnEffectiveSizeDropPolicy, TriggerOnDropBelow30Percent) {
  auto policy = beluga::policies::on_effective_size_drop(0.3);
  auto weights1 = std::vector{1.0, 1.0, 1.0, 1.0, 1.0};
  auto weights2 = std::vector{1.0, 0.0, 0.0, 0.0, 0.0};  // Fewer effective particles
 
  ASSERT_FALSE(policy(weights1));  // Initial ESS should not trigger the policy
  ASSERT_TRUE(policy(weights2));   // ESS drops below 30% on the second call, triggering the policy
}
 
TEST(OnEffectiveSizeDropPolicy, NoTriggerAbove30Percent) {
  auto policy = beluga::policies::on_effective_size_drop(0.3);
  auto weights1 = std::vector{1.0, 1.0, 1.0, 1.0, 1.0};
  auto weights2 = std::vector{1.0, 1.0, 0.0, 0.0, 0.0};  // Still above 30%
 
  ASSERT_FALSE(policy(weights1));  // Initial ESS should not trigger the policy
  ASSERT_FALSE(policy(weights1));  // ESS remains above 30%, no trigger
}
 
TEST(OnEffectiveSizeDropPolicy, TriggerOnDropBelowDefaultThreshold) {
  auto policy = beluga::policies::on_effective_size_drop;
  auto weights = std::vector{1.0, 1.0, 0.0, 0.0, 0.0};
 
  ASSERT_TRUE(policy(weights));  // ESS drops below 50%
}
 
}  // namespace