X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=test%2FTestHungarian.cpp;h=3d6173fdcfa54041d412b22d6174350ff270e16d;hb=3ff9a5ad691b8dca9d91f8e9786a8d08d31b70fa;hp=f20ec09245227cf7568b4fe23e98463840d29ab5;hpb=0d625404c2ef4a286e1feee4485ff60c0fb9730e;p=trackerpp.git

diff --git a/test/TestHungarian.cpp b/test/TestHungarian.cpp
index f20ec09..3d6173f 100644
--- a/test/TestHungarian.cpp
+++ b/test/TestHungarian.cpp
@@ -1,5 +1,6 @@
 #include "hungarian.h"
 #include "gtest/gtest.h"
+#include <cmath>
 
 using namespace std;
 using namespace Eigen;
@@ -44,3 +45,36 @@ TEST(Hungarian, 4x3)
     EXPECT_TRUE(expect_row_ind == row_ind);
     EXPECT_TRUE(expect_col_ind == col_ind);
 }
+
+TEST(Distance, consine)
+{
+    Vector3d u, v;
+    u << 1, 0, 0;
+    v << 0, 1, 0;
+    double d = distance_cosine(u, v);
+    EXPECT_DOUBLE_EQ(d, 1.0);
+
+    u << 100, 0, 0;
+    v << 0, 1, 0;
+    d = distance_cosine(u, v);
+    EXPECT_DOUBLE_EQ(d, 1.0);
+
+    u << 1, 1, 0;
+    v << 0, 1, 0;
+    d = distance_cosine(u, v);
+    EXPECT_TRUE(std::abs(d - 0.2928932) < 0.0001);
+}
+
+TEST(Distance, euclidean)
+{
+    Vector3d u, v;
+    u << 1, 0, 0;
+    v << 0, 1, 0;
+    double d = distance_euclidean(u, v);
+    EXPECT_TRUE(std::abs(d - 1.41421356) < 0.0001);
+
+    u << 1, 1, 0;
+    v << 0, 1, 0;
+    d = distance_euclidean(u, v);
+    EXPECT_DOUBLE_EQ(d, 1.0);
+}