[mlpack] 07/53: First test case.

[Barak A. Pearlmutter]

This is an automated email from the git hooks/post-receive script.

bap pushed a commit to branch master
in repository mlpack.

commit 443028c17710e77caea5eeddbcdc2419791c8c45
Author: Ryan Curtin <ryan at ratml.org>
Date:   Wed Apr 13 13:04:53 2016 -0700

    First test case.
---
 qdafn_test.cpp | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/qdafn_test.cpp b/qdafn_test.cpp
new file mode 100644
index 0000000..83c1da4
--- /dev/null
+++ b/qdafn_test.cpp
@@ -0,0 +1,50 @@
+/**
+ * @file qdafn_test.cpp
+ * @author Ryan Curtin
+ *
+ * Test the QDAFN functionality.
+ */
+#define BOOST_TEST_MODULE QDAFNTest
+
+#include <boost/test/unit_test.hpp>
+
+#include <mlpack/core.hpp>
+#include "qdafn.hpp"
+
+using namespace std;
+using namespace arma;
+using namespace mlpack;
+using namespace qdafn;
+
+/**
+ * With one reference point, make sure that is the one that is returned.
+ */
+BOOST_AUTO_TEST_CASE(QDAFNTrivialTest)
+{
+  arma::mat refSet(5, 1);
+  refSet.randu();
+
+  // 5 tables, 1 point.
+  QDAFN<> qdafn(refSet, 5, 1);
+
+  arma::mat querySet(5, 5);
+  querySet.randu();
+
+  arma::Mat<size_t> neighbors;
+  arma::mat distances;
+  qdafn.Search(querySet, 1, neighbors, distances);
+
+  // Check sizes.
+  BOOST_REQUIRE_EQUAL(neighbors.n_rows, 1);
+  BOOST_REQUIRE_EQUAL(neighbors.n_cols, 5);
+  BOOST_REQUIRE_EQUAL(distances.n_rows, 1);
+  BOOST_REQUIRE_EQUAL(distances.n_cols, 5);
+
+  for (size_t i = 0; i < 5; ++i)
+  {
+    BOOST_REQUIRE_EQUAL(neighbors[i], 0);
+    const double dist = metric::EuclideanDistance::Evaluate(querySet.col(i),
+        refSet.col(0));
+    BOOST_REQUIRE_CLOSE(distances[i], dist, 1e-5);
+  }
+}

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/mlpack.git