From 0ae0b1a9a433aa548f68e91c984e39a5216c4fa7 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Fri, 25 Sep 2015 22:53:31 +0000
Subject: [PATCH] Refactor tests for changed NBC API.

No new tests for Train() or empty constructor yet.
---
 src/mlpack/tests/nbc_test.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/tests/nbc_test.cpp b/src/mlpack/tests/nbc_test.cpp
index a8b6ad7414..4fdec14295 100644
--- a/src/mlpack/tests/nbc_test.cpp
+++ b/src/mlpack/tests/nbc_test.cpp
@@ -27,7 +27,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierTest)
   data::Load(trainResultFilename, trainRes, true);
 
   // Get the labels out.
-  arma::Col<size_t> labels(trainData.n_cols);
+  arma::Row<size_t> labels(trainData.n_cols);
   for (size_t i = 0; i < trainData.n_cols; ++i)
     labels[i] = trainData(trainData.n_rows - 1, i);
   trainData.shed_row(trainData.n_rows - 1);
@@ -55,7 +55,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierTest)
 
   arma::mat testData;
   arma::Mat<size_t> testRes;
-  arma::Col<size_t> calcVec;
+  arma::Row<size_t> calcVec;
   data::Load(testFilename, testData, true);
   data::Load(testResultFilename, testRes, true);
 
@@ -82,7 +82,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierIncrementalTest)
   data::Load(trainResultFilename, trainRes, true);
 
   // Get the labels out.
-  arma::Col<size_t> labels(trainData.n_cols);
+  arma::Row<size_t> labels(trainData.n_cols);
   for (size_t i = 0; i < trainData.n_cols; ++i)
     labels[i] = trainData(trainData.n_rows - 1, i);
   trainData.shed_row(trainData.n_rows - 1);
@@ -110,7 +110,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierIncrementalTest)
 
   arma::mat testData;
   arma::Mat<size_t> testRes;
-  arma::Col<size_t> calcVec;
+  arma::Row<size_t> calcVec;
   data::Load(testFilename, testData, true);
   data::Load(testResultFilename, testRes, true);