Check for invalid data. (#6742)

2021-03-04 14:37:20 +08:00
parent a9b4a95225
commit f20074e826
2 changed files with 19 additions and 6 deletions
--- a/tests/cpp/data/test_simple_dmatrix.cc
+++ b/tests/cpp/data/test_simple_dmatrix.cc
@@ -116,6 +116,14 @@ TEST(SimpleDMatrix, MissingData) {
  CHECK_EQ(dmat->Info().num_nonzero_, 2);
  dmat.reset(new data::SimpleDMatrix(&adapter, 1.0, 1));
  CHECK_EQ(dmat->Info().num_nonzero_, 1);
+
+  {
+    data[1] = std::numeric_limits<float>::infinity();
+    data::DenseAdapter adapter(data.data(), data.size(), 1);
+    EXPECT_THROW(data::SimpleDMatrix dmat(
+                     &adapter, std::numeric_limits<float>::quiet_NaN(), -1),
+                 dmlc::Error);
+  }
 }

 TEST(SimpleDMatrix, EmptyRow) {