Add unit test for callback generated by the user.

shogun-toolbox · May 25, 2018 · 8d720dd · 8d720dd
1 parent da950d5
commit 8d720dd
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/unit/lib/StoppableSGObject_unittest.cc b/tests/unit/lib/StoppableSGObject_unittest.cc
@@ -73,3 +73,24 @@ TEST(StoppableSGObject, custom_callback)
 	a.train();
 	EXPECT_TRUE(a.get_check() == 5);
 }
+
+TEST(StoppableSGObject, custom_callback_by_user)
+{
+	int i=0;
+	function<bool()> callback = [&i]()
+	{
+		if (i>=3) {
+			get_global_signal()->get_subscriber()->on_next(SG_BLOCK_COMP);
+			return true;
+		}
+		i++;
+		return false;
+	};
+
+
+	Mock_model a;
+	a.add_callback(callback);
+	a.train();
+	EXPECT_TRUE(a.get_check() == 3);
+}
+