1//===----------------------------------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is dual licensed under the MIT and the University of Illinois Open
6// Source Licenses. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10// <random>
11
12// template<class RealType = double>
13// class piecewise_linear_distribution
14
15// template<class InputIterator>
16//     piecewise_linear_distribution(InputIteratorB firstB,
17//                                     InputIteratorB lastB,
18//                                     InputIteratorW firstW);
19
20#include <random>
21#include <cassert>
22
23int main()
24{
25    {
26        typedef std::piecewise_linear_distribution<> D;
27        double b[] = {10};
28        double p[] = {12};
29        D d(b, b, p);
30        std::vector<double> iv = d.intervals();
31        assert(iv.size() == 2);
32        assert(iv[0] == 0);
33        assert(iv[1] == 1);
34        std::vector<double> dn = d.densities();
35        assert(dn.size() == 2);
36        assert(dn[0] == 1);
37        assert(dn[1] == 1);
38    }
39    {
40        typedef std::piecewise_linear_distribution<> D;
41        double b[] = {10};
42        double p[] = {12};
43        D d(b, b+1, p);
44        std::vector<double> iv = d.intervals();
45        assert(iv.size() == 2);
46        assert(iv[0] == 0);
47        assert(iv[1] == 1);
48        std::vector<double> dn = d.densities();
49        assert(dn.size() == 2);
50        assert(dn[0] == 1);
51        assert(dn[1] == 1);
52    }
53    {
54        typedef std::piecewise_linear_distribution<> D;
55        double b[] = {10, 15};
56        double p[] = {20, 20};
57        D d(b, b+2, p);
58        std::vector<double> iv = d.intervals();
59        assert(iv.size() == 2);
60        assert(iv[0] == 10);
61        assert(iv[1] == 15);
62        std::vector<double> dn = d.densities();
63        assert(dn.size() == 2);
64        assert(dn[0] == 1/5.);
65        assert(dn[1] == 1/5.);
66    }
67    {
68        typedef std::piecewise_linear_distribution<> D;
69        double b[] = {10, 15, 16};
70        double p[] = {.25, .75, .25};
71        D d(b, b+3, p);
72        std::vector<double> iv = d.intervals();
73        assert(iv.size() == 3);
74        assert(iv[0] == 10);
75        assert(iv[1] == 15);
76        assert(iv[2] == 16);
77        std::vector<double> dn = d.densities();
78        assert(dn.size() == 3);
79        assert(dn[0] == .25/3);
80        assert(dn[1] == .75/3);
81        assert(dn[2] == .25/3);
82    }
83    {
84        typedef std::piecewise_linear_distribution<> D;
85        double b[] = {10, 14, 16, 17};
86        double p[] = {0, 1, 1, 0};
87        D d(b, b+4, p);
88        std::vector<double> iv = d.intervals();
89        assert(iv.size() == 4);
90        assert(iv[0] == 10);
91        assert(iv[1] == 14);
92        assert(iv[2] == 16);
93        assert(iv[3] == 17);
94        std::vector<double> dn = d.densities();
95        assert(dn.size() == 4);
96        assert(dn[0] == 0);
97        assert(dn[1] == 1/4.5);
98        assert(dn[2] == 1/4.5);
99        assert(dn[3] == 0);
100    }
101}
102