41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
/* ----------------------------------------------------------------------------
|
|
|
|
* 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
|
|
|
|
* -------------------------------------------------------------------------- */
|
|
|
|
/**
|
|
* @file testWeightedSampler.cpp
|
|
* @brief Unit test for WeightedSampler
|
|
* @author Frank Dellaert
|
|
* @date MAy 2019
|
|
**/
|
|
|
|
#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);
|
|
}
|