test_statistics.cpp

// Description: Testing for statistics
// DocumentationOf: statistics.h

#include "test/test_init.h"

#include <pastel/sys/rational.h>
#include <pastel/sys/locator.h>
#include <pastel/sys/pointset.h>
#include <pastel/sys/set.h>

#include <pastel/math/statistic.h>

#include <vector>
#include <list>

namespace
{

    using Real = Rational<integer>;

}

TEST_CASE("statistics (statistics)")
{
    {
        using Point = Real;
        std::vector<Point> aSet = { -1, 7, 4, -3, 7, 8 };
        std::vector<Point> bSet = { -6, 4, 3, 2, -6, -1 };

        using PointSet = decltype(aSet);
        PASTEL_CONCEPT_CHECK(PointSet, PointSet_Concept);

        REQUIRE(allEqual(pointMean(aSet), Real(11, 3)));
        REQUIRE(allEqual(pointVariance(aSet, PASTEL_TAG(biased), false), Real(322, 15)));
        REQUIRE(allEqual(pointVariance(aSet), Real(322, 18)));

        REQUIRE(allEqual(pointMean(bSet), Real(-2, 3)));
        REQUIRE(allEqual(pointVariance(bSet, PASTEL_TAG(biased), false), Real(298, 15)));
        REQUIRE(allEqual(pointVariance(bSet), Real(298, 18)));

        REQUIRE(scalarCovariance<Real>(aSet, bSet, PASTEL_TAG(biased), false) == Real(14, 15));
        REQUIRE(scalarMeanSquareError<Real>(aSet, bSet) == Real(310, 6));
    }
    {
        std::vector<Real> aSet = { -1, 7, 4, -3, 7, 8 };
        std::vector<Real> bSet = { -6, 4, 3, 2, -6, -1 };

        REQUIRE(scalarMean<Real>(aSet) == Real(11, 3));
        REQUIRE(scalarVariance<Real>(aSet, PASTEL_TAG(biased), false) == Real(322, 15));
        REQUIRE(scalarVariance<Real>(aSet) == Real(322, 18));

        REQUIRE(scalarMean<Real>(bSet) == Real(-2, 3));
        REQUIRE(scalarVariance<Real>(bSet, PASTEL_TAG(biased), false) == Real(298, 15));
        REQUIRE(scalarVariance<Real>(bSet) == Real(298, 18));

        REQUIRE(scalarCovariance<Real>(aSet, bSet, PASTEL_TAG(biased), false) == Real(14, 15));
        REQUIRE(scalarMeanSquareError<Real>(aSet, bSet) == Real(310, 6));
    }
    /*
   {
       using Point = Vector<Real, 2>;
       using Locator = Vector_Locator<Real, 2>;
       std::list<Point> aSet = { { -1, -6 }, { 7, 4 }, { 4, 3 }, { -3, 2 }, { 7, -6 }, { 8, -1 } };

       REQUIRE(pointMean(locationSet(rangeInput(aSet), Locator())) == Point(Real(11, 3), Real(-2, 3)));
       REQUIRE(pointVariance(locationSet(rangeInput(aSet), Locator()), false) == Point(Real(322, 15), Real(298, 15)));
       REQUIRE(pointVariance(locationSet(rangeInput(aSet), Locator())) == Point(Real(322, 18), Real(298, 18)));
   }
   */
}