[mlpack-git] master: A first simple test. (1beb58d)

[gitdub at mlpack.org]

Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/ee95e2030ddd526368e377b4e1c13484d471e307...a2440f163ac76edda36a9f7ba31cfecd3a55654e

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

commit 1beb58dbb90bf2349c2ac495669103e61231b628
Author: Ryan Curtin <ryan at ratml.org>
Date:   Tue May 31 14:27:39 2016 -0400

    A first simple test.


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

1beb58dbb90bf2349c2ac495669103e61231b628
 src/mlpack/tests/load_save_test.cpp | 40 +++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/src/mlpack/tests/load_save_test.cpp b/src/mlpack/tests/load_save_test.cpp
index 7e2f6a1..fc753f8 100644
--- a/src/mlpack/tests/load_save_test.cpp
+++ b/src/mlpack/tests/load_save_test.cpp
@@ -1094,6 +1094,46 @@ BOOST_AUTO_TEST_CASE(CategoricalNontransposedCSVLoadTest)
 }
 
 /**
+ * The test CSV Keon suggested in #658.
+ */
+BOOST_AUTO_TEST_CASE(KeonsSimpleDatasetInfoTest)
+{
+  fstream f;
+  f.open("test.csv", fstream::out);
+  f << "1, 1, 1" << endl;
+  f << ", 1, 1" << endl;
+  f << "1, 1, 1" << endl;
+  f << "1, 1, 1" << endl;
+  f.close();
+
+  // Load without transpose.
+  arma::mat dataset;
+  data::DatasetInfo info;
+  data::Load("test.csv", dataset, info, true, false);
+
+  BOOST_REQUIRE_EQUAL(dataset.n_rows, 4);
+  BOOST_REQUIRE_EQUAL(dataset.n_cols, 3);
+  BOOST_REQUIRE_EQUAL(info.Dimensionality(), 4);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(0), 0);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(1), 2);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(2), 0);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(3), 0);
+
+  // Load with transpose.
+  data::DatasetInfo newInfo;
+  data::Load("test.csv", dataset, info, true, true);
+
+  BOOST_REQUIRE_EQUAL(dataset.n_rows, 3);
+  BOOST_REQUIRE_EQUAL(dataset.n_cols, 4);
+  BOOST_REQUIRE_EQUAL(info.Dimensionality(), 3);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(0), 2);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(1), 0);
+  BOOST_REQUIRE_EQUAL(info.NumMappings(2), 0);
+
+  remove("test.csv");
+}
+
+/**
  * A simple ARFF load test.  Two attributes, both numeric.
  */
 BOOST_AUTO_TEST_CASE(SimpleARFFTest)