summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/accumulators/test/covariance.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/boost/libs/accumulators/test/covariance.cpp')
-rw-r--r--src/boost/libs/accumulators/test/covariance.cpp135
1 files changed, 135 insertions, 0 deletions
diff --git a/src/boost/libs/accumulators/test/covariance.cpp b/src/boost/libs/accumulators/test/covariance.cpp
new file mode 100644
index 00000000..abdbc9e8
--- /dev/null
+++ b/src/boost/libs/accumulators/test/covariance.cpp
@@ -0,0 +1,135 @@
+// (C) Copyright 2005 Daniel Egloff, Eric Niebler
+// Use, modification and distribution are subject to the
+// Boost Software License, Version 1.0. (See accompanying file
+// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+
+#define BOOST_NUMERIC_FUNCTIONAL_STD_VECTOR_SUPPORT
+
+#include <boost/test/unit_test.hpp>
+#include <boost/test/floating_point_comparison.hpp>
+#include <boost/accumulators/accumulators.hpp>
+#include <boost/accumulators/statistics/variates/covariate.hpp>
+#include <boost/accumulators/statistics/stats.hpp>
+#include <boost/accumulators/statistics/covariance.hpp>
+#include <sstream>
+#include <boost/archive/text_oarchive.hpp>
+#include <boost/archive/text_iarchive.hpp>
+
+using namespace boost;
+using namespace unit_test;
+using namespace accumulators;
+
+///////////////////////////////////////////////////////////////////////////////
+// test_stat
+//
+void test_stat()
+{
+ std::vector<double> dummy;
+ dummy.push_back(0);
+ dummy.push_back(0);
+
+ accumulator_set<double, stats<tag::covariance<double, tag::covariate1> > > acc;
+ accumulator_set<std::vector<double>, stats<tag::covariance<double, tag::covariate1> > > acc2(sample = dummy);
+ accumulator_set<double, stats<tag::covariance<std::vector<double>, tag::covariate1> > > acc3(covariate1 = dummy);
+ accumulator_set<std::vector<double>, stats<tag::covariance<std::vector<double>, tag::covariate1> > > acc4(sample = dummy, covariate1 = dummy);
+
+ std::vector<double> a;
+ a.push_back(1.);
+ a.push_back(2.);
+ std::vector<double> b;
+ b.push_back(3.);
+ b.push_back(4.);
+ std::vector<double> c;
+ c.push_back(2.);
+ c.push_back(5.);
+ std::vector<double> d;
+ d.push_back(4.);
+ d.push_back(2.);
+
+ // double - double
+ {
+ acc(1., covariate1 = 2.);
+ acc(1., covariate1 = 4.);
+ acc(2., covariate1 = 3.);
+ acc(6., covariate1 = 1.);
+ }
+
+ // vector - double
+ {
+ acc2(a, covariate1 = 1.);
+ acc2(b, covariate1 = 1.);
+ acc2(c, covariate1 = 2.);
+ acc2(d, covariate1 = 6.);
+ }
+
+ // double - vector
+ {
+ acc3(1., covariate1 = a);
+ acc3(1., covariate1 = b);
+ acc3(2., covariate1 = c);
+ acc3(6., covariate1 = d);
+ }
+
+ // vector - vector
+ {
+ acc4(a, covariate1 = b);
+ acc4(b, covariate1 = c);
+ acc4(a, covariate1 = c);
+ acc4(d, covariate1 = b);
+ }
+
+ double epsilon = 1e-6;
+
+ BOOST_CHECK_CLOSE((covariance(acc)), -1.75, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc2))[0], 1.75, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc2))[1], -1.125, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc3))[0], 1.75, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc3))[1], -1.125, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc4))(0,0), 0.125, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc4))(0,1), -0.25, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc4))(1,0), -0.125, epsilon);
+ BOOST_CHECK_CLOSE((covariance(acc4))(1,1), 0.25, epsilon);
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// test_persistency
+//
+void test_persistency()
+{
+ // "persistent" storage
+ std::stringstream ss;
+ // tolerance
+ double epsilon = 1e-6;
+ {
+
+ accumulator_set<double, stats<tag::covariance<double, tag::covariate1> > > acc;
+ acc(1., covariate1 = 2.);
+ acc(1., covariate1 = 4.);
+ acc(2., covariate1 = 3.);
+ acc(6., covariate1 = 1.);
+
+
+ BOOST_CHECK_CLOSE((covariance(acc)), -1.75, epsilon);
+ boost::archive::text_oarchive oa(ss);
+ acc.serialize(oa, 0);
+ }
+ accumulator_set<double, stats<tag::covariance<double, tag::covariate1> > > acc;
+ BOOST_CHECK_CLOSE((covariance(acc)), 0., epsilon);
+ boost::archive::text_iarchive ia(ss);
+ acc.serialize(ia, 0);
+ BOOST_CHECK_CLOSE((covariance(acc)), -1.75, epsilon);
+
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// init_unit_test_suite
+//
+test_suite* init_unit_test_suite( int argc, char* argv[] )
+{
+ test_suite *test = BOOST_TEST_SUITE("covariance test");
+
+ test->add(BOOST_TEST_CASE(&test_stat));
+ test->add(BOOST_TEST_CASE(&test_persistency));
+
+ return test;
+}