From df2dac2a7ea42152181cf5e805f0b70ecd205853 Mon Sep 17 00:00:00 2001 From: Ryan Curtin Date: Fri, 25 Sep 2015 17:28:04 -0400 Subject: [PATCH] Add unfinished serialization test. --- src/mlpack/tests/serialization_test.cpp | 28 +++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/src/mlpack/tests/serialization_test.cpp b/src/mlpack/tests/serialization_test.cpp index 3f2d70ed31..061e79909c 100644 --- a/src/mlpack/tests/serialization_test.cpp +++ b/src/mlpack/tests/serialization_test.cpp @@ -24,6 +24,7 @@ #include #include +#include using namespace mlpack; using namespace mlpack::distribution; @@ -745,4 +746,31 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionTest) BOOST_REQUIRE_CLOSE(lr.Lambda(), lrBinary.Lambda(), 1e-5); } +BOOST_AUTO_TEST_CASE(AllkNNTest) +{ + using neighbor::AllkNN; + arma::mat dataset = arma::randu(5, 2000); +/* + AllkNN allknn(dataset, false, false); + + AllkNN knnXml, knnText, knnBinary; + + SerializeObjectAll(allknn, knnXml, knnText, knnBinary); + + // Now run nearest neighbor and make sure the results are the same. + arma::mat querySet = arma::randu(5, 1000); + + arma::mat distances, xmlDistances, textDistances, binaryDistances; + arma::Mat neighbors, xmlNeighbors, textNeighbors, binaryNeighbors; + + allknn.Search(querySet, 5, neighbors, distances); + knnXml.Search(querySet, 5, xmlNeighbors, xmlDistances); + knnText.Search(querySet, 5, textNeighbors, textDistances); + knnBinary.Search(querySet, 5, binaryNeighbors, binaryDistances); + + CheckMatrices(distances, xmlDistances, textDistances, binaryDistances); + CheckMatrices(neighbors, xmlNeighbors, textNeighbors, binaryNeighbors); +*/ +} + BOOST_AUTO_TEST_SUITE_END();