1 | // (C) Copyright Eric Niebler 2005. |
2 | // Use, modification and distribution are subject to the |
3 | // Boost Software License, Version 1.0. (See accompanying file |
4 | // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
5 | |
6 | #include <boost/test/unit_test.hpp> |
7 | #include <boost/test/tools/floating_point_comparison.hpp> |
8 | #include <boost/accumulators/accumulators.hpp> |
9 | #include <boost/accumulators/statistics/stats.hpp> |
10 | #include <boost/accumulators/statistics/weighted_mean.hpp> |
11 | |
12 | using namespace boost; |
13 | using namespace unit_test; |
14 | using namespace accumulators; |
15 | |
16 | /////////////////////////////////////////////////////////////////////////////// |
17 | // test_stat |
18 | // |
19 | void test_stat() |
20 | { |
21 | accumulator_set<int, stats<tag::weighted_mean>, external<int> > acc; |
22 | accumulator_set<void, stats<tag::sum_of_weights>, int> weight_acc; |
23 | |
24 | acc(10, weight = 2); // 20 |
25 | weight_acc(weight = 2); // |
26 | BOOST_CHECK_EQUAL(2, sum_of_weights(weight_acc)); // |
27 | // |
28 | acc(6, weight = 3); // 18 |
29 | weight_acc(weight = 3); // |
30 | BOOST_CHECK_EQUAL(5, sum_of_weights(weight_acc)); // |
31 | // |
32 | acc(4, weight = 4); // 16 |
33 | weight_acc(weight = 4); // |
34 | BOOST_CHECK_EQUAL(9, sum_of_weights(weight_acc)); // |
35 | // |
36 | acc(6, weight = 5); //+ 30 |
37 | weight_acc(weight = 5); // |
38 | BOOST_CHECK_EQUAL(14, sum_of_weights(weight_acc)); // |
39 | //= 84 / 14 = 6 |
40 | |
41 | BOOST_CHECK_CLOSE(6., weighted_mean(acc, weights = weight_acc), 1e-5); |
42 | } |
43 | |
44 | /////////////////////////////////////////////////////////////////////////////// |
45 | // init_unit_test_suite |
46 | // |
47 | test_suite* init_unit_test_suite( int argc, char* argv[] ) |
48 | { |
49 | test_suite *test = BOOST_TEST_SUITE("external_weights test" ); |
50 | |
51 | test->add(BOOST_TEST_CASE(&test_stat)); |
52 | |
53 | return test; |
54 | } |
55 | |