From c0d18d251769be8c24362fefb0b68e18df857a6f Mon Sep 17 00:00:00 2001
From: xiefangqi <xiefangqi2@huawei.com>
Date: Thu, 20 Aug 2020 11:01:17 +0800
Subject: [PATCH] fix randomdata ut problem

---
 tests/ut/cpp/dataset/c_api_dataset_randomdata_test.cc | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/tests/ut/cpp/dataset/c_api_dataset_randomdata_test.cc b/tests/ut/cpp/dataset/c_api_dataset_randomdata_test.cc
index 1e986cc9b3..22e77a2ddc 100644
--- a/tests/ut/cpp/dataset/c_api_dataset_randomdata_test.cc
+++ b/tests/ut/cpp/dataset/c_api_dataset_randomdata_test.cc
@@ -100,11 +100,8 @@ TEST_F(MindDataTestPipeline, TestRandomDatasetBasic2) {
   // Check if RandomDataOp read correct columns
   uint64_t i = 0;
   while (row.size() != 0) {
-    auto image = row["image"];
-    auto label = row["label"];
-    MS_LOG(INFO) << "Tensor image shape: " << image->shape();
-    MS_LOG(INFO) << "Tensor label shape: " << label->shape();
-
+    // If no schema specified, RandomData will generate random columns
+    // So we don't check columns here
     iter->GetNextRow(&row);
     i++;
   }