test/proximity_list_test.cpp

   1 #include "../src/util/proximity_list.hpp"
   2 #define BOOST_TEST_MODULE ProximityListTest
   3 #include <boost/test/unit_test.hpp>
   4 #include <boost/bind.hpp>
   5 #include <boost/random.hpp>
   6 #include <algorithm>
   7 #include <cmath>
   8 #include <iostream>
   9 #include <vector>
  10
  11 struct Distance1
  12 {
  13     typedef float result_type;
  14
  15     template<class T>
  16     result_type operator()(const T &left, const T &right) const
  17     {
  18         return std::abs(left - right);
  19     }
  20 };
  21
  22 BOOST_AUTO_TEST_CASE(constructors)
  23 {
  24     proximity_list<int, Distance1> p1;
  25
  26     BOOST_CHECK(p1.begin() == p1.end());
  27
  28     std::vector<int> vec;
  29     for(size_t i = 0; i < 10; ++i)
  30     {
  31         vec.push_back(i);
  32     }
  33
  34     proximity_list<int, Distance1> p2(vec.begin(), vec.end());
  35
  36     BOOST_CHECK_EQUAL(vec.size(), p2.size());
  37
  38 }
  39 BOOST_AUTO_TEST_CASE(insert)
  40 {
  41     proximity_list<int, Distance1> p1;
  42     p1.insert(1);
  43     p1.insert(2);
  44     p1.insert(3);
  45
  46     BOOST_CHECK_EQUAL(p1.size(), 3);
  47
  48     std::vector<int> vec;
  49     for(size_t i = 0; i < 10; ++i)
  50     {
  51         vec.push_back(i);
  52     }
  53     p1.insert(vec.begin(), vec.end());
  54
  55     BOOST_CHECK_EQUAL(p1.size(), 13);
  56 }
  57 BOOST_AUTO_TEST_CASE(find_nearest)
  58 {
  59     boost::mt19937 rng;
  60     boost::uniform_int<> size(0, 100);
  61     boost::variate_generator<boost::mt19937 &, boost::uniform_int<> > random(
  62         rng, size);
  63
  64     std::vector<int> vec;
  65     for(size_t i = 0; i < 10; ++i)
  66     {
  67         vec.push_back(random());
  68     }
  69     proximity_list<int, Distance1> p1(vec.begin(), vec.end());
  70
  71     for(size_t i = 0; i < 100; ++i)
  72     {
  73         int query = random();
  74         int nearest1 = *p1.find_nearest(query);
  75         int nearest2 = *std::max_element(vec.begin(), vec.end(), boost::bind(
  76             std::less<Distance1::result_type>(),
  77             boost::bind(Distance1(), _2, query),
  78             boost::bind(Distance1(), _1, query)));
  79
  80         BOOST_CHECK_EQUAL(nearest1, nearest2);
  81     }
  82 }