c@391: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
c@391: 
c@391: #include "maths/MedianFilter.h"
c@391: 
c@391: #include <cmath>
c@391: 
c@391: #define BOOST_TEST_DYN_LINK
c@391: #define BOOST_TEST_MAIN
c@391: 
c@391: #include <boost/test/unit_test.hpp>
c@391: 
c@391: BOOST_AUTO_TEST_SUITE(TestMedianFilter)
c@391: 
c@391: BOOST_AUTO_TEST_CASE(odd)
c@391: {
c@391:     // A median filter of size N always retains a pool of N elements,
c@391:     // which start out all-zero. So the output median will remain zero
c@391:     // until N/2 elements have been pushed.
c@391:     MedianFilter<double> f(3);
c@391:     f.push(1);			// 0 0 1
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(-3);			// 0 1 -3
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(5);			// 1 -3 5
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     f.push(7);			// -3 5 7
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// 5 7 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// 7 3 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 3);
c@391: }
c@391: 
c@391: BOOST_AUTO_TEST_CASE(even)
c@391: {
c@391:     // Our median does not halve the difference (should it??), it just
c@391:     // returns the next element from the input set
c@391:     MedianFilter<double> f(4);
c@391:     f.push(1);			// 0 0 0 1
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(-3);			// 0 0 1 -3
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(5);			// 0 1 -3 5
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     f.push(7);			// 1 -3 5 7
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// -3 5 7 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// 5 7 3 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391: }
c@391: 
c@391: BOOST_AUTO_TEST_CASE(odd75)
c@391: {
c@391:     MedianFilter<double> f(5, 75.f);
c@391:     f.push(1);			// 0 0 0 0 1
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(-3);			// 0 0 0 1 -3
c@391:     BOOST_CHECK_EQUAL(f.get(), 0);
c@391:     f.push(5);			// 0 0 1 -3 5
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     f.push(7);			// 0 1 -3 5 7
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// 0 -3 5 7 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(3);			// 0 5 7 3 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391: }
c@391: 
c@391: BOOST_AUTO_TEST_CASE(even75)
c@391: {
c@391:     MedianFilter<double> f(4, 75.f);
c@391:     f.push(1);			// 0 0 0 1
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     f.push(-3);			// 0 0 1 -3
c@391:     BOOST_CHECK_EQUAL(f.get(), 1);
c@391:     f.push(5);			// 0 1 -3 5
c@391:     BOOST_CHECK_EQUAL(f.get(), 5);
c@391:     f.push(7);			// 1 -3 5 7
c@391:     BOOST_CHECK_EQUAL(f.get(), 7);
c@391:     BOOST_CHECK_EQUAL(f.get(), 7);
c@391:     f.push(3);			// -3 5 7 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 7);
c@391:     f.push(3);			// 5 7 3 3
c@391:     BOOST_CHECK_EQUAL(f.get(), 7);
c@391: }
c@391: 
c@391: BOOST_AUTO_TEST_SUITE_END()
c@391: 
c@391: