test/TestHungarian.cpp

   1 #include "hungarian.h"
   2 #include "gtest/gtest.h"
   3
   4 using namespace std;
   5 using namespace Eigen;
   6
   7 TEST(Hungarian, Verify)
   8 {
   9     Matrix3i C;
  10     C << 1, 2, 3,
  11          2, 4, 2,
  12          3, 6, 9;
  13
  14     VectorXi row_ind, col_ind;
  15     int ret = linear_sum_assignment(C, row_ind, col_ind);
  16     Vector3i expect_row_ind, expect_col_ind;
  17
  18     expect_row_ind << 0, 1, 2;
  19     expect_col_ind << 1, 2, 0;
  20
  21     EXPECT_EQ(ret, 7);
  22     EXPECT_TRUE(expect_row_ind == row_ind);
  23     EXPECT_TRUE(expect_col_ind == col_ind);
  24 }