1 // (C) Copyright Eric Niebler, Olivier Gygi 2006.
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/accumulators/accumulators.hpp>
8 #include <boost/accumulators/statistics/stats.hpp>
9 #include <boost/accumulators/statistics/weighted_sum.hpp>
10 #include <boost/accumulators/statistics/variates/covariate.hpp>
11
12 using namespace boost;
13 using namespace unit_test;
14 using namespace accumulators;
15
16 ///////////////////////////////////////////////////////////////////////////////
17 // test_stat
18 //
test_stat()19 void test_stat()
20 {
21 accumulator_set<int, stats<tag::weighted_sum, tag::weighted_sum_of_variates<int, tag::covariate1> >, int> acc;
22
23 acc(1, weight = 2, covariate1 = 3);
24 BOOST_CHECK_EQUAL(2, weighted_sum(acc));
25 BOOST_CHECK_EQUAL(6, weighted_sum_of_variates(acc));
26
27 acc(2, weight = 3, covariate1 = 6);
28 BOOST_CHECK_EQUAL(8, weighted_sum(acc));
29 BOOST_CHECK_EQUAL(24, weighted_sum_of_variates(acc));
30
31 acc(4, weight = 6, covariate1 = 9);
32 BOOST_CHECK_EQUAL(32, weighted_sum(acc));
33 BOOST_CHECK_EQUAL(78, weighted_sum_of_variates(acc));
34 }
35
36 ///////////////////////////////////////////////////////////////////////////////
37 // init_unit_test_suite
38 //
init_unit_test_suite(int argc,char * argv[])39 test_suite* init_unit_test_suite( int argc, char* argv[] )
40 {
41 test_suite *test = BOOST_TEST_SUITE("weighted_sum test");
42
43 test->add(BOOST_TEST_CASE(&test_stat));
44
45 return test;
46 }
47