testWeightedSampler.cpp

Go to the documentation of this file.

/* ----------------------------------------------------------------------------
 
 * GTSAM Copyright 2010, Georgia Tech Research Corporation,
 * Atlanta, Georgia 30332-0415
 * All Rights Reserved
 * Authors: Frank Dellaert, et al. (see THANKS for the full author list)
 
 * See LICENSE for the license information
 
 * -------------------------------------------------------------------------- */
 
#include <gtsam/base/WeightedSampler.h>
 
#include <CppUnitLite/TestHarness.h>
 
#include <random>
 
using namespace std;
using namespace gtsam;
 
TEST(WeightedSampler, sampleWithoutReplacement) {
  vector<double> weights{1, 2, 3, 4, 3, 2, 1};
  std::mt19937 rng(42);
  WeightedSampler<std::mt19937> sampler(&rng);
  auto samples = sampler.sampleWithoutReplacement(5, weights);
  EXPECT_LONGS_EQUAL(5, samples.size());
}
 
//******************************************************************************
int main() {
  TestResult tr;
  return TestRegistry::runAllTests(tr);
}