1//===----------------------------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9// UNSUPPORTED: c++03
10
11// <random>
12
13// template<class IntType = int>
14// class discrete_distribution
15
16// param_type(initializer_list<double> wl);
17
18#include <random>
19
20#include <cassert>
21#include <vector>
22
23#include "test_macros.h"
24
25int main(int, char**)
26{
27 {
28 typedef std::discrete_distribution<> D;
29 typedef D::param_type P;
30 P pa = {};
31 std::vector<double> p = pa.probabilities();
32 assert(p.size() == 1);
33 assert(p[0] == 1);
34 }
35 {
36 typedef std::discrete_distribution<> D;
37 typedef D::param_type P;
38 P pa = {10};
39 std::vector<double> p = pa.probabilities();
40 assert(p.size() == 1);
41 assert(p[0] == 1);
42 }
43 {
44 typedef std::discrete_distribution<> D;
45 typedef D::param_type P;
46 P pa = {10, 30};
47 std::vector<double> p = pa.probabilities();
48 assert(p.size() == 2);
49 assert(p[0] == 0.25);
50 assert(p[1] == 0.75);
51 }
52 {
53 typedef std::discrete_distribution<> D;
54 typedef D::param_type P;
55 P pa = {30, 10};
56 std::vector<double> p = pa.probabilities();
57 assert(p.size() == 2);
58 assert(p[0] == 0.75);
59 assert(p[1] == 0.25);
60 }
61 {
62 typedef std::discrete_distribution<> D;
63 typedef D::param_type P;
64 P pa = {30, 0, 10};
65 std::vector<double> p = pa.probabilities();
66 assert(p.size() == 3);
67 assert(p[0] == 0.75);
68 assert(p[1] == 0);
69 assert(p[2] == 0.25);
70 }
71 {
72 typedef std::discrete_distribution<> D;
73 typedef D::param_type P;
74 P pa = {0, 30, 10};
75 std::vector<double> p = pa.probabilities();
76 assert(p.size() == 3);
77 assert(p[0] == 0);
78 assert(p[1] == 0.75);
79 assert(p[2] == 0.25);
80 }
81 {
82 typedef std::discrete_distribution<> D;
83 typedef D::param_type P;
84 P pa = {0, 0, 10};
85 std::vector<double> p = pa.probabilities();
86 assert(p.size() == 3);
87 assert(p[0] == 0);
88 assert(p[1] == 0);
89 assert(p[2] == 1);
90 }
91
92 return 0;
93}
94

source code of libcxx/test/std/numerics/rand/rand.dist/rand.dist.samp/rand.dist.samp.discrete/param_ctor_init.pass.cpp