35 lines
842 B
C++
35 lines
842 B
C++
|
#include "tap.hpp"
|
||
|
|
|
||
|
|
#include "kmeans.hpp"
|
||
|
|
|
||
|
|
#include <cruft/util/point.hpp>
|
||
|
|
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
int
|
||
|
|
main ()
|
||
|
|
{
|
||
|
|
util::TAP::logger tap;
|
||
|
|
|
||
|
|
// create one point and check it 'converges' to this one point
|
||
|
|
{
|
||
|
|
const std::array<util::point3f,1> p { {{1,2,3}} };
|
||
|
|
std::array<util::point3f,1> q;
|
||
|
|
|
||
|
|
util::kmeans (util::view{p}, util::view{q});
|
||
|
|
tap.expect_eq (p, q, "single point, single k");
|
||
|
|
}
|
||
|
|
|
||
|
|
// create two vectors, check if the mean converges to their average
|
||
|
|
{
|
||
|
|
const std::array<util::vector3f,2> p {{
|
||
|
|
{1}, {2}
|
||
|
|
}};
|
||
|
|
std::array<util::vector3f,1> q;
|
||
|
|
|
||
|
|
util::kmeans (util::view{p}, util::view{q});
|
||
|
|
tap.expect_eq (q[0], (p[0]+p[1])/2, "two point, single k");
|
||
|
|
}
|
||
|
|
|
||
|
|
return tap.status ();
|
||
|
|
}
|