Add callback test for lmnn

2019-11-03 11:24:16 +05:30
parent d81c298b9c
commit ea401b7dd3
1 changed files with 4 additions and 1 deletions
@@ -357,7 +357,10 @@ BOOST_AUTO_TEST_CASE(LMNNSGDSimpleDatasetTest)
  LMNN<> lmnn(dataset, labels, 1);

  arma::mat outputMatrix;
-  lmnn.LearnDistance(outputMatrix);
+  std::stringstream stream;
+
+  lmnn.LearnDistance(outputMatrix, ens::ProgressBar(70, stream));
+  BOOST_REQUIRE_GT(stream.str().length(), 0);

  // Ensure that the objective function is better now.
  LMNNFunction<> lmnnfn(dataset, labels, 1, 0.6, 1);