diff --git a/src/include/units/random.h b/src/include/units/random.h new file mode 100644 index 00000000..8570e1e9 --- /dev/null +++ b/src/include/units/random.h @@ -0,0 +1,453 @@ +#pragma once + +#include +#include + +namespace units { + +template +struct uniform_int_distribution : public std::uniform_int_distribution +{ + using Rep = Quantity::rep; + using Base = std::uniform_int_distribution; + + uniform_int_distribution() : Base() {} + uniform_int_distribution(Quantity a, Quantity b) : Base(a.count(), b.count()) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity a() const { return Quantity(Base::a()); } + Quantity b() const { return Quantity(Base::b()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct uniform_real_distribution : public std::uniform_real_distribution +{ + using Rep = Quantity::rep; + using Base = std::uniform_real_distribution; + + uniform_real_distribution() : Base() {} + uniform_real_distribution(Quantity a, Quantity b) : Base(a.count(), b.count()) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity a() const { return Quantity(Base::a()); } + Quantity b() const { return Quantity(Base::b()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct binomial_distribution : public std::binomial_distribution +{ + using Rep = Quantity::rep; + using Base = std::binomial_distribution; + + binomial_distribution() : Base() {} + binomial_distribution(Quantity t, double p) : Base(t.count(), p) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity t() const { return Quantity(Base::t()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct negative_binomial_distribution : public std::negative_binomial_distribution +{ + using Rep = Quantity::rep; + using Base = std::negative_binomial_distribution; + + negative_binomial_distribution() : Base() {} + negative_binomial_distribution(Quantity k, double p) : Base(k.count(), p) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity k() const { return Quantity(Base::k()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct geometric_distribution : public std::geometric_distribution +{ + using Rep = Quantity::rep; + using Base = std::geometric_distribution; + + geometric_distribution() : Base() {} + geometric_distribution(double p) : Base(p) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct poisson_distribution : public std::poisson_distribution +{ + using Rep = Quantity::rep; + using Base = std::poisson_distribution; + + poisson_distribution() : Base() {} + poisson_distribution(double p) : Base(p) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct exponential_distribution : public std::exponential_distribution +{ + using Rep = Quantity::rep; + using Base = std::exponential_distribution; + + exponential_distribution() : Base() {} + exponential_distribution(Rep lambda) : Base(lambda) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct gamma_distribution : public std::gamma_distribution +{ + using Rep = Quantity::rep; + using Base = std::gamma_distribution; + + gamma_distribution() : Base() {} + gamma_distribution(Rep alpha, Rep beta) : Base(alpha, beta) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct weibull_distribution : public std::weibull_distribution +{ + using Rep = Quantity::rep; + using Base = std::weibull_distribution; + + weibull_distribution() : Base() {} + weibull_distribution(Rep a, Rep b) : Base(a, b) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct extreme_value_distribution : public std::extreme_value_distribution +{ + using Rep = Quantity::rep; + using Base = std::extreme_value_distribution; + + extreme_value_distribution() : Base() {} + extreme_value_distribution(Quantity a, Rep b) : Base(a.count(), b) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity a() const { return Quantity(Base::a()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct normal_distribution : public std::normal_distribution +{ + using Rep = Quantity::rep; + using Base = std::normal_distribution; + + normal_distribution() : Base() {} + normal_distribution(Quantity mean, Quantity stddev) : Base(mean.count(), stddev.count()) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity mean() const { return Quantity(Base::mean()); } + Quantity stddev() const { return Quantity(Base::stddev()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct lognormal_distribution : public std::lognormal_distribution +{ + using Rep = Quantity::rep; + using Base = std::lognormal_distribution; + + lognormal_distribution() : Base() {} + lognormal_distribution(Quantity m, Quantity s) : Base(m.count(), s.count()) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity m() const { return Quantity(Base::m()); } + Quantity s() const { return Quantity(Base::s()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct chi_squared_distribution : public std::chi_squared_distribution +{ + using Rep = Quantity::rep; + using Base = std::chi_squared_distribution; + + chi_squared_distribution() : Base() {} + chi_squared_distribution(Rep n) : Base(n) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct cauchy_distribution : public std::cauchy_distribution +{ + using Rep = Quantity::rep; + using Base = std::cauchy_distribution; + + cauchy_distribution() : Base() {} + cauchy_distribution(Quantity a, Quantity b) : Base(a.count(), b.count()) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity a() const { return Quantity(Base::a()); } + Quantity b() const { return Quantity(Base::b()); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct fisher_f_distribution : public std::fisher_f_distribution +{ + using Rep = Quantity::rep; + using Base = std::fisher_f_distribution; + + fisher_f_distribution() : Base() {} + fisher_f_distribution(Rep m, Rep n) : Base(m, n) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct student_t_distribution : public std::student_t_distribution +{ + using Rep = Quantity::rep; + using Base = std::student_t_distribution; + + student_t_distribution() : Base() {} + student_t_distribution(Rep n) : Base(n) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +struct discrete_distribution : public std::discrete_distribution +{ + using Rep = Quantity::rep; + using Base = std::discrete_distribution; + + discrete_distribution() : Base() {} + + template + discrete_distribution(InputIt first, InputIt last) : Base(first, last) {} + + discrete_distribution(std::initializer_list weights) : Base(weights) {} + + template + discrete_distribution(std::size_t count, double xmin, double xmax, UnaryOperation unary_op) : + Base(count, xmin, xmax, unary_op) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +class piecewise_constant_distribution : public std::piecewise_constant_distribution +{ + using Rep = Quantity::rep; + using Base = std::piecewise_constant_distribution; + + template + inline static std::vector i_qty_to_rep(InputIt first, InputIt last) + { + std::vector intervals_rep; + intervals_rep.reserve(static_cast(std::distance(first, last))); + for (auto itr = first; itr != last; ++itr) { intervals_rep.push_back(itr->count()); } + return intervals_rep; + } + + inline static std::vector bl_qty_to_rep(std::initializer_list& bl) + { + std::vector bl_rep; + bl_rep.reserve(bl.size()); + for (const Quantity& qty : bl) { bl_rep.push_back(qty.count()); } + return bl_rep; + } + + template + inline static std::vector fw_bl(std::initializer_list& bl, UnaryOperation fw) + { + std::vector w_bl; + w_bl.reserve(bl.size()); + for (const Quantity& qty : bl) { w_bl.push_back(fw(qty)); } + std::vector weights; + weights.reserve(bl.size()); + for (size_t i = 0; i < bl.size() - 1; ++i) { weights.push_back(w_bl[i] + w_bl[i + 1]); } + weights.push_back(0); + return weights; + } + + template + piecewise_constant_distribution(const std::vector& i, InputIt first_w) : + Base(i.cbegin(), i.cend(), first_w) {} + + piecewise_constant_distribution(const std::vector& bl, const std::vector& weights) : + Base(bl.cbegin(), bl.cend(), weights.cbegin()) {} + +public: + piecewise_constant_distribution() : Base() {} + + template + piecewise_constant_distribution(InputIt1 first_i, InputIt1 last_i, InputIt2 first_w) : + piecewise_constant_distribution(i_qty_to_rep(first_i, last_i), first_w) {} + + template + piecewise_constant_distribution(std::initializer_list bl, UnaryOperation fw) : + piecewise_constant_distribution(bl_qty_to_rep(bl), fw_bl(bl, fw)) {} + + template + piecewise_constant_distribution(std::size_t nw, Quantity xmin, Quantity xmax, UnaryOperation fw) : + Base(nw, xmin.count(), xmax.count(), [fw](Rep val) { return fw(Quantity(val)); }) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + std::vector intervals() const + { + std::vector intervals_rep = Base::intervals(); + std::vector intervals_qty; + intervals_qty.reserve(intervals_rep.size()); + for (const Rep& val : intervals_rep) { intervals_qty.push_back(Quantity(val)); } + return intervals_qty; + } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +template +class piecewise_linear_distribution : public std::piecewise_linear_distribution +{ + using Rep = Quantity::rep; + using Base = std::piecewise_linear_distribution; + + template + inline static std::vector i_qty_to_rep(InputIt first, InputIt last) + { + std::vector intervals_rep; + intervals_rep.reserve(static_cast(std::distance(first, last))); + for (auto itr = first; itr != last; ++itr) { intervals_rep.push_back(itr->count()); } + return intervals_rep; + } + + inline static std::vector bl_qty_to_rep(std::initializer_list& bl) + { + std::vector bl_rep; + bl_rep.reserve(bl.size()); + for (const Quantity& qty : bl) { bl_rep.push_back(qty.count()); } + return bl_rep; + } + + template + inline static std::vector fw_bl(std::initializer_list& bl, UnaryOperation fw) + { + std::vector weights; + weights.reserve(bl.size()); + for (const Quantity& qty : bl) { weights.push_back(fw(qty)); } + return weights; + } + + template + piecewise_linear_distribution(const std::vector& i, InputIt first_w) : + Base(i.cbegin(), i.cend(), first_w) {} + + piecewise_linear_distribution(const std::vector& bl, const std::vector& weights) : + Base(bl.cbegin(), bl.cend(), weights.cbegin()) {} + +public: + piecewise_linear_distribution() : Base() {} + + template + piecewise_linear_distribution(InputIt1 first_i, InputIt1 last_i, InputIt2 first_w) : + piecewise_linear_distribution(i_qty_to_rep(first_i, last_i), first_w) {} + + template + piecewise_linear_distribution(std::initializer_list bl, UnaryOperation fw) : + piecewise_linear_distribution(bl_qty_to_rep(bl), fw_bl(bl, fw)) {} + + template + piecewise_linear_distribution(std::size_t nw, Quantity xmin, Quantity xmax, UnaryOperation fw) : + Base(nw, xmin.count(), xmax.count(), [fw](Rep val) { return fw(Quantity(val)); }) {} + + template + Quantity operator()(Generator& g) { return Quantity(Base::operator()(g)); } + + std::vector intervals() const + { + std::vector intervals_rep = Base::intervals(); + std::vector intervals_qty; + intervals_qty.reserve(intervals_rep.size()); + for (const Rep& val : intervals_rep) { intervals_qty.push_back(Quantity(val)); } + return intervals_qty; + } + + Quantity min() const { return Quantity(Base::min()); } + Quantity max() const { return Quantity(Base::max()); } +}; + +} // namespace units \ No newline at end of file diff --git a/test/unit_test/runtime/CMakeLists.txt b/test/unit_test/runtime/CMakeLists.txt index f1cb289b..098ece1f 100644 --- a/test/unit_test/runtime/CMakeLists.txt +++ b/test/unit_test/runtime/CMakeLists.txt @@ -29,6 +29,7 @@ add_executable(unit_tests_runtime math_test.cpp fmt_test.cpp fmt_units_test.cpp + distribution_test.cpp ) target_link_libraries(unit_tests_runtime PRIVATE diff --git a/test/unit_test/runtime/distribution_test.cpp b/test/unit_test/runtime/distribution_test.cpp new file mode 100644 index 00000000..37e74792 --- /dev/null +++ b/test/unit_test/runtime/distribution_test.cpp @@ -0,0 +1,587 @@ +#include +#include +#include +#include + +using namespace units; +using namespace units::physical::si; + +TEST_CASE("uniform_int_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto dist = units::uniform_int_distribution(); + + CHECK(dist.a() == Qty::zero()); + CHECK(dist.b() == Qty::max()); + } + + SECTION ("parametrized") { + constexpr Rep a = 5; + constexpr Rep b = 2; + + auto stl_dist = std::uniform_int_distribution(a, b); + auto units_dist = units::uniform_int_distribution(Qty(a), Qty(b)); + + CHECK(units_dist.a() == Qty(stl_dist.a())); + CHECK(units_dist.b() == Qty(stl_dist.b())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("uniform_real_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::uniform_real_distribution(); + + CHECK(dist.a() == Qty::zero()); + CHECK(dist.b() == Qty::one()); + } + + SECTION ("parametrized") { + constexpr Rep a = 5.0; + constexpr Rep b = 2.0; + + auto stl_dist = std::uniform_real_distribution(a, b); + auto units_dist = units::uniform_real_distribution(Qty(a), Qty(b)); + + CHECK(units_dist.a() == Qty(stl_dist.a())); + CHECK(units_dist.b() == Qty(stl_dist.b())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("binomial_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto dist = units::binomial_distribution(); + + CHECK(dist.p() == 0.5); + CHECK(dist.t() == Qty::one()); + } + + SECTION ("parametrized") { + constexpr Rep t = 5; + constexpr double p = 0.25; + + auto stl_dist = std::binomial_distribution(t, p); + auto units_dist = units::binomial_distribution(Qty(t), p); + + CHECK(units_dist.p() == stl_dist.p()); + CHECK(units_dist.t() == Qty(stl_dist.t())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("negative_binomial_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto dist = units::negative_binomial_distribution(); + + CHECK(dist.p() == 0.5); + CHECK(dist.k() == Qty::one()); + } + + SECTION ("parametrized") { + constexpr Rep k = 5; + constexpr double p = 0.25; + + auto stl_dist = std::negative_binomial_distribution(k, p); + auto units_dist = units::negative_binomial_distribution(Qty(k), p); + + CHECK(units_dist.p() == stl_dist.p()); + CHECK(units_dist.k() == Qty(stl_dist.k())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("geometric_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto dist = units::geometric_distribution(); + + CHECK(dist.p() == 0.5); + } + + SECTION ("parametrized") { + constexpr double p = 0.25; + + auto stl_dist = std::geometric_distribution(p); + auto units_dist = units::geometric_distribution(p); + + CHECK(units_dist.p() == stl_dist.p()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("poisson_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto dist = units::poisson_distribution(); + + CHECK(dist.mean() == 1.0); + } + + SECTION ("parametrized") { + constexpr double mean = 5.0; + + auto stl_dist = std::poisson_distribution(mean); + auto units_dist = units::poisson_distribution(mean); + + CHECK(units_dist.mean() == stl_dist.mean()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("exponential_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::exponential_distribution(); + + CHECK(dist.lambda() == 1.0); + } + + SECTION ("parametrized") { + constexpr double lambda = 2.0; + + auto stl_dist = std::exponential_distribution(lambda); + auto units_dist = units::exponential_distribution(lambda); + + CHECK(units_dist.lambda() == stl_dist.lambda()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("gamma_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::gamma_distribution(); + + CHECK(dist.alpha() == 1.0); + CHECK(dist.beta() == 1.0); + } + + SECTION ("parametrized") { + constexpr double alpha = 5.0; + constexpr double beta = 2.0; + + auto stl_dist = std::gamma_distribution(alpha, beta); + auto units_dist = units::gamma_distribution(alpha, beta); + + CHECK(units_dist.alpha() == stl_dist.alpha()); + CHECK(units_dist.beta() == stl_dist.beta()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("weibull_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::weibull_distribution(); + + CHECK(dist.a() == 1.0); + CHECK(dist.b() == 1.0); + } + + SECTION ("parametrized") { + constexpr Rep a = 5.0; + constexpr Rep b = 2.0; + + auto stl_dist = std::weibull_distribution(a, b); + auto units_dist = units::weibull_distribution(a, b); + + CHECK(units_dist.a() == stl_dist.a()); + CHECK(units_dist.b() == stl_dist.b()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("extreme_value_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::extreme_value_distribution(); + + CHECK(dist.a() == Qty::zero()); + CHECK(dist.b() == 1.0); + } + + SECTION ("parametrized") { + constexpr Rep a = 5.0; + constexpr Rep b = 2.0; + + auto stl_dist = std::extreme_value_distribution(a, b); + auto units_dist = units::extreme_value_distribution(Qty(a), b); + + CHECK(units_dist.a() == Qty(stl_dist.a())); + CHECK(units_dist.b() == stl_dist.b()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("normal_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::normal_distribution(); + + CHECK(dist.mean() == Qty::zero()); + CHECK(dist.stddev() == Qty::one()); + } + + SECTION("parametrized") + { + constexpr Rep mean = 5.0; + constexpr Rep stddev = 2.0; + + auto stl_dist = std::normal_distribution(mean, stddev); + auto units_dist = units::normal_distribution(Qty(mean), Qty(stddev)); + + CHECK(units_dist.mean() == Qty(stl_dist.mean())); + CHECK(units_dist.stddev() == Qty(stl_dist.stddev())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("lognormal_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::lognormal_distribution(); + + CHECK(dist.m() == Qty::zero()); + CHECK(dist.s() == Qty::one()); + } + + SECTION("parametrized") + { + constexpr Rep m = 5.0; + constexpr Rep s = 2.0; + + auto stl_dist = std::lognormal_distribution(m, s); + auto units_dist = units::lognormal_distribution(Qty(m), Qty(s)); + + CHECK(units_dist.m() == Qty(stl_dist.m())); + CHECK(units_dist.s() == Qty(stl_dist.s())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("chi_squared_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::chi_squared_distribution(); + + CHECK(dist.n() == 1.0); + } + + SECTION("parametrized") + { + constexpr Rep n = 5.0; + + auto stl_dist = std::chi_squared_distribution(n); + auto units_dist = units::chi_squared_distribution(n); + + CHECK(units_dist.n() == stl_dist.n()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("cauchy_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::cauchy_distribution(); + + CHECK(dist.a() == Qty::zero()); + CHECK(dist.b() == Qty::one()); + } + + SECTION ("parametrized") { + constexpr Rep a = 5.0; + constexpr Rep b = 2.0; + + auto stl_dist = std::cauchy_distribution(a, b); + auto units_dist = units::cauchy_distribution(Qty(a), Qty(b)); + + CHECK(units_dist.a() == Qty(stl_dist.a())); + CHECK(units_dist.b() == Qty(stl_dist.b())); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("fisher_f_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::fisher_f_distribution(); + + CHECK(dist.m() == 1.0); + CHECK(dist.n() == 1.0); + } + + SECTION ("parametrized") { + constexpr Rep m = 5.0; + constexpr Rep n = 2.0; + + auto stl_dist = std::fisher_f_distribution(m, n); + auto units_dist = units::fisher_f_distribution(m, n); + + CHECK(units_dist.m() == stl_dist.m()); + CHECK(units_dist.n() == stl_dist.n()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("student_t_distribution") +{ + using Rep = long double; + using Qty = length; + + SECTION("default") + { + auto dist = units::student_t_distribution(); + + CHECK(dist.n() == 1.0); + } + + SECTION ("parametrized") { + constexpr Rep n = 2.0; + + auto stl_dist = std::student_t_distribution(n); + auto units_dist = units::student_t_distribution(n); + + CHECK(units_dist.n() == stl_dist.n()); + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + } +} + +TEST_CASE("discrete_distribution") +{ + using Rep = std::int64_t; + using Qty = length; + + SECTION("default") + { + auto stl_dist = std::discrete_distribution(); + auto units_dist = units::discrete_distribution(); + + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + CHECK(units_dist.probabilities() == stl_dist.probabilities()); + } + + SECTION ("parametrized_input_it") { + constexpr std::array weights = {1.0, 2.0, 3.0}; + + auto stl_dist = std::discrete_distribution(weights.cbegin(), weights.cend()); + auto units_dist = units::discrete_distribution(weights.cbegin(), weights.cend()); + + CHECK(units_dist.probabilities() == stl_dist.probabilities()); + } + + SECTION ("parametrized_initializer_list") { + std::initializer_list weights = {1.0, 2.0, 3.0}; + + auto stl_dist = std::discrete_distribution(weights); + auto units_dist = units::discrete_distribution(weights); + + CHECK(units_dist.probabilities() == stl_dist.probabilities()); + } + + SECTION ("parametrized_range") { + constexpr std::size_t count = 3; + constexpr double xmin = 1, xmax = 3; + + auto stl_dist = std::discrete_distribution(count, xmin, xmax, [](double val) { return val; }); + auto units_dist = units::discrete_distribution(count, xmin, xmax, [](double val) { return val; }); + + CHECK(units_dist.probabilities() == stl_dist.probabilities()); + } +} + +TEST_CASE("piecewise_constant_distribution") +{ + using Rep = long double; + using Qty = length; + + std::vector intervals_rep_vec = {1.0, 2.0, 3.0}; + std::vector intervals_qty_vec = {1.0q_m, 2.0q_m, 3.0q_m}; + + SECTION("default") + { + auto stl_dist = std::piecewise_constant_distribution(); + auto units_dist = units::piecewise_constant_distribution(); + + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + CHECK(stl_dist.intervals().size() == 2); + CHECK(units_dist.intervals().size() == 2); + CHECK(stl_dist.densities().size() == 1); + CHECK(units_dist.densities().size() == 1); + } + + SECTION ("parametrized_input_it") { + constexpr std::array intervals_rep = {1.0, 2.0, 3.0}; + constexpr std::array intervals_qty = {1.0q_m, 2.0q_m, 3.0q_m}; + constexpr std::array weights = {1.0, 2.0, 3.0}; + + auto stl_dist = std::piecewise_constant_distribution(intervals_rep.cbegin(), intervals_rep.cend(), weights.cbegin()); + auto units_dist = units::piecewise_constant_distribution(intervals_qty.cbegin(), intervals_qty.cend(), weights.cbegin()); + + CHECK(stl_dist.intervals() == intervals_rep_vec); + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } + + SECTION ("parametrized_initializer_list") { + std::initializer_list intervals_rep = {1.0, 2.0, 3.0}; + std::initializer_list intervals_qty = {1.0q_m, 2.0q_m, 3.0q_m}; + + auto stl_dist = std::piecewise_constant_distribution(intervals_rep, [](Rep val) { return val; }); + auto units_dist = units::piecewise_constant_distribution(intervals_qty, [](Qty qty) { return qty.count(); }); + + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } + + SECTION ("parametrized_range") { + constexpr std::size_t nw = 2; + constexpr Rep xmin_rep = 1.0, xmax_rep = 3.0; + constexpr Qty xmin_qty = 1.0q_m, xmax_qty = 3.0q_m; + + auto stl_dist = std::piecewise_constant_distribution(nw, xmin_rep, xmax_rep, [](Rep val) { return val; }); + auto units_dist = units::piecewise_constant_distribution(nw, xmin_qty, xmax_qty, [](Qty qty) { return qty.count(); }); + + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } +} + +TEST_CASE("piecewise_linear_distribution") +{ + using Rep = long double; + using Qty = length; + + std::vector intervals_rep_vec = {1.0, 2.0, 3.0}; + std::vector intervals_qty_vec = {1.0q_m, 2.0q_m, 3.0q_m}; + + SECTION("default") + { + auto stl_dist = std::piecewise_linear_distribution(); + auto units_dist = units::piecewise_linear_distribution(); + + CHECK(units_dist.min() == Qty(stl_dist.min())); + CHECK(units_dist.max() == Qty(stl_dist.max())); + CHECK(stl_dist.intervals().size() == 2); + CHECK(units_dist.intervals().size() == 2); + CHECK(stl_dist.densities().size() == 2); + CHECK(units_dist.densities().size() == 2); + } + + SECTION ("parametrized_input_it") { + constexpr std::array intervals_rep = {1.0, 2.0, 3.0}; + constexpr std::array intervals_qty = {1.0q_m, 2.0q_m, 3.0q_m}; + constexpr std::array weights = {1.0, 2.0, 3.0}; + + auto stl_dist = std::piecewise_linear_distribution(intervals_rep.cbegin(), intervals_rep.cend(), weights.cbegin()); + auto units_dist = units::piecewise_linear_distribution(intervals_qty.cbegin(), intervals_qty.cend(), weights.cbegin()); + + CHECK(stl_dist.intervals() == intervals_rep_vec); + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } + + SECTION ("parametrized_initializer_list") { + std::initializer_list intervals_rep = {1.0, 2.0, 3.0}; + std::initializer_list intervals_qty = {1.0q_m, 2.0q_m, 3.0q_m}; + + auto stl_dist = std::piecewise_linear_distribution(intervals_rep, [](Rep val) { return val; }); + auto units_dist = units::piecewise_linear_distribution(intervals_qty, [](Qty qty) { return qty.count(); }); + + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } + + SECTION ("parametrized_range") { + constexpr std::size_t nw = 2; + constexpr Rep xmin_rep = 1.0, xmax_rep = 3.0; + constexpr Qty xmin_qty = 1.0q_m, xmax_qty = 3.0q_m; + + auto stl_dist = std::piecewise_linear_distribution(nw, xmin_rep, xmax_rep, [](Rep val) { return val; }); + auto units_dist = units::piecewise_linear_distribution(nw, xmin_qty, xmax_qty, [](Qty qty) { return qty.count(); }); + + CHECK(units_dist.intervals() == intervals_qty_vec); + CHECK(units_dist.densities() == stl_dist.densities()); + } +} \ No newline at end of file