1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
//===----------------------------------------------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is dual licensed under the MIT and the University of Illinois Open
// Source Licenses. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
// <random>
// template<class IntType = int>
// class discrete_distribution
// discrete_distribution(initializer_list<double> wl);
#include <random>
#include <cassert>
int main()
{
#ifndef _LIBCPP_HAS_NO_GENERALIZED_INITIALIZERS
{
typedef std::discrete_distribution<> D;
D d = {};
std::vector<double> p = d.probabilities();
assert(p.size() == 1);
assert(p[0] == 1);
}
{
typedef std::discrete_distribution<> D;
D d = {10};
std::vector<double> p = d.probabilities();
assert(p.size() == 1);
assert(p[0] == 1);
}
{
typedef std::discrete_distribution<> D;
D d = {10, 30};
std::vector<double> p = d.probabilities();
assert(p.size() == 2);
assert(p[0] == 0.25);
assert(p[1] == 0.75);
}
{
typedef std::discrete_distribution<> D;
D d = {30, 10};
std::vector<double> p = d.probabilities();
assert(p.size() == 2);
assert(p[0] == 0.75);
assert(p[1] == 0.25);
}
{
typedef std::discrete_distribution<> D;
D d = {30, 0, 10};
std::vector<double> p = d.probabilities();
assert(p.size() == 3);
assert(p[0] == 0.75);
assert(p[1] == 0);
assert(p[2] == 0.25);
}
{
typedef std::discrete_distribution<> D;
D d = {0, 30, 10};
std::vector<double> p = d.probabilities();
assert(p.size() == 3);
assert(p[0] == 0);
assert(p[1] == 0.75);
assert(p[2] == 0.25);
}
{
typedef std::discrete_distribution<> D;
D d = {0, 0, 10};
std::vector<double> p = d.probabilities();
assert(p.size() == 3);
assert(p[0] == 0);
assert(p[1] == 0);
assert(p[2] == 1);
}
#endif // _LIBCPP_HAS_NO_GENERALIZED_INITIALIZERS
}
|