[mlpack-git] master: fix tests for split data (ac016fe)

[gitdub at mlpack.org]

Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/e3a23c256f017ebb8185b15847c82f51d359cdfd...181792d99549467440b2b839f52deec75be10334

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

commit ac016feb8fb38b0b5c20644adff628815a4dc880
Author: Keon Kim <kwk236 at gmail.com>
Date:   Thu Jun 2 16:27:14 2016 +0900

    fix tests for split data


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

ac016feb8fb38b0b5c20644adff628815a4dc880
 src/mlpack/tests/split_data_test.cpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/tests/split_data_test.cpp b/src/mlpack/tests/split_data_test.cpp
index daf4cd5..bbc529b 100644
--- a/src/mlpack/tests/split_data_test.cpp
+++ b/src/mlpack/tests/split_data_test.cpp
@@ -36,7 +36,10 @@ void CompareData(const mat& inputData,
     const mat& rhsCol = compareData.col(i);
     for (size_t j = 0; j != lhsCol.n_rows; ++j)
     {
-      BOOST_REQUIRE_CLOSE(lhsCol(j), rhsCol(j), 1e-5);
+      if (std::abs(rhsCol(j)) < 1e-5)
+        BOOST_REQUIRE_SMALL(lhsCol(j), 1e-5);
+      else
+        BOOST_REQUIRE_CLOSE(lhsCol(j), rhsCol(j), 1e-5);
     }
   }
 }
@@ -52,7 +55,10 @@ void CheckMatEqual(const mat& inputData,
     const mat& rhsCol = sortedCompare.col(i);
     for (size_t j = 0; j < lhsCol.n_rows; ++j)
     {
-      BOOST_REQUIRE_CLOSE(lhsCol(j), rhsCol(j), 1e-5);
+      if (std::abs(rhsCol(j)) < 1e-5)
+        BOOST_REQUIRE_SMALL(lhsCol(j), 1e-5);
+      else
+        BOOST_REQUIRE_CLOSE(lhsCol(j), rhsCol(j), 1e-5);
     }
   }
 }