From adb3040bf13ce42d2c0ade36347c43315f1a91fa Mon Sep 17 00:00:00 2001
From: Vikas Shetty <shettyvikas209@gmail.com>
Date: Sun, 20 Oct 2019 17:28:47 +0530
Subject: [PATCH] Adding test for callback

---
 src/mlpack/tests/callback_test.cpp | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/src/mlpack/tests/callback_test.cpp b/src/mlpack/tests/callback_test.cpp
index d0190c2b36..b1d8fe3e5c 100644
--- a/src/mlpack/tests/callback_test.cpp
+++ b/src/mlpack/tests/callback_test.cpp
@@ -5,11 +5,13 @@
 #include <mlpack/methods/ann/ffn.hpp>
 #include <mlpack/methods/ann/rnn.hpp>
 #include <mlpack/methods/ann/loss_functions/mean_squared_error.hpp>
+#include <mlpack/methods/logistic_regression/logistic_regression.hpp>
 
 #include <boost/test/unit_test.hpp>
 
 using namespace mlpack;
 using namespace mlpack::ann;
+using namespace mlpack::regression;
 
 BOOST_AUTO_TEST_SUITE(CallbackTest);
 
@@ -115,4 +117,26 @@ BOOST_AUTO_TEST_CASE(RNNWithOptimizerCallbackTest)
   BOOST_REQUIRE_GT(stream.str().length(), 0);
 }
 
+/**
+ *  Test Logistic regression implementation with Printloss callback.
+ */
+
+BOOST_AUTO_TEST_CASE(LRWithOptimizerCallback)
+{
+    arma::mat data("1 2 3;"
+                   "1 2 3");
+    arma::Row<size_t> responses("1 1 0");
+    ens::StandardSGD sgd(0.005, 32, 500000, 1e-4);
+    LogisticRegression<> logisticRegression(data, responses, sgd, 0.001);
+    std::stringstream stream;
+    logisticRegression.Train<ens::L_BFGS>(
+            data,
+            responses,
+            sgd,
+            ens::PrintLoss(stream)
+            );
+    BOOST_REQUIRE_GT(stream.str().length(), 0);
+
+}
+
 BOOST_AUTO_TEST_SUITE_END();