2 #include "gtest/gtest.h"
15 VectorXi row_ind, col_ind;
16 int ret = linear_sum_assignment(C, row_ind, col_ind);
17 cout << "row: [" << row_ind.transpose() << "], col: [" << col_ind.transpose() << "]" << endl;
18 Vector3i expect_row_ind, expect_col_ind;
20 expect_row_ind << 0, 1, 2;
21 expect_col_ind << 2, 1, 0;
24 EXPECT_TRUE(expect_row_ind == row_ind);
25 EXPECT_TRUE(expect_col_ind == col_ind);
37 VectorXi row_ind, col_ind;
38 int ret = linear_sum_assignment(C, row_ind, col_ind);
39 Vector3i expect_row_ind, expect_col_ind;
41 expect_row_ind << 0, 1, 3;
42 expect_col_ind << 1, 2, 0;
45 EXPECT_TRUE(expect_row_ind == row_ind);
46 EXPECT_TRUE(expect_col_ind == col_ind);
49 TEST(Distance, consine)
54 double d = distance_cosine(u, v);
55 EXPECT_DOUBLE_EQ(d, 1.0);
59 d = distance_cosine(u, v);
60 EXPECT_DOUBLE_EQ(d, 1.0);
64 d = distance_cosine(u, v);
65 EXPECT_TRUE(std::abs(d - 0.2928932) < 0.0001);
68 TEST(Distance, euclidean)
73 double d = distance_euclidean(u, v);
74 EXPECT_TRUE(std::abs(d - 1.41421356) < 0.0001);
78 d = distance_euclidean(u, v);
79 EXPECT_DOUBLE_EQ(d, 1.0);