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);
}