[mlpack-git] master, mlpack-1.0.x: Decision Stump test fixed (796acc7)

[gitdub at big.cc.gt.atl.ga.us]

Repository : https://github.com/mlpack/mlpack

On branches: master,mlpack-1.0.x
Link       : https://github.com/mlpack/mlpack/compare/904762495c039e345beba14c1142fd719b3bd50e...f94823c800ad6f7266995c700b1b630d5ffdcf40

>---------------------------------------------------------------

commit 796acc7e08a0075cf1e460d52114624f1a9c9fc3
Author: Udit Saxena <saxena.udit at gmail.com>
Date:   Tue Jun 24 18:17:55 2014 +0000

    Decision Stump test fixed


>---------------------------------------------------------------

796acc7e08a0075cf1e460d52114624f1a9c9fc3
 src/mlpack/tests/decision_stump_test.cpp | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/tests/decision_stump_test.cpp b/src/mlpack/tests/decision_stump_test.cpp
index c826295..34970a1 100644
--- a/src/mlpack/tests/decision_stump_test.cpp
+++ b/src/mlpack/tests/decision_stump_test.cpp
@@ -75,7 +75,11 @@ BOOST_AUTO_TEST_CASE(PerfectSplitOnZero)
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,4),1);
 }
 
 /**
@@ -103,7 +107,7 @@ BOOST_AUTO_TEST_CASE(BinningTesting)
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
 }
 
 /**
@@ -134,7 +138,10 @@ BOOST_AUTO_TEST_CASE(PerfectMultiClassSplit)
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),2);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),3);
 }
 
 /**
@@ -167,7 +174,14 @@ BOOST_AUTO_TEST_CASE(MultiClassSplit)
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,4),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,5),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,6),2);
+  BOOST_CHECK_EQUAL(predictedLabels(0,7),2);
 }
 
 BOOST_AUTO_TEST_SUITE_END();