/** * Copyright 2020 Huawei Technologies Co., Ltd * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include "minddata/dataset/util/circular_pool.h" #include "minddata/dataset/core/client.h" #include "minddata/dataset/engine/jagged_connector.h" #include "common/common.h" #include "gtest/gtest.h" #include "utils/log_adapter.h" using namespace mindspore::dataset; using mindspore::LogStream; using mindspore::ExceptionType::NoExceptionType; using mindspore::MsLogLevel::INFO; class MindDataTestSkipOp : public UT::DatasetOpTesting {}; TEST_F(MindDataTestSkipOp, TestSkipOpFuntions) { // Start with an empty execution tree auto my_tree = std::make_shared(); Status rc; std::string dataset_path; dataset_path = datasets_root_path_ + "/testTFTestAllTypes/test.data"; std::shared_ptr config_manager = GlobalContext::config_manager(); int32_t op_connector_size = config_manager->op_connector_size(); int32_t num_workers = config_manager->num_parallel_workers(); int32_t worker_connector_size = 16; std::unique_ptr schema = std::make_unique(); schema->LoadSchemaFile(datasets_root_path_ + "/testTFTestAllTypes/datasetSchema.json", {}); std::vector columns_to_load = {}; std::vector files = {dataset_path}; std::shared_ptr my_tfreader_op = std::make_shared(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); rc = my_tfreader_op->Init(); ASSERT_TRUE(rc.IsOk()); rc = my_tree->AssociateNode(my_tfreader_op); ASSERT_TRUE(rc.IsOk()); // SkipOp std::shared_ptr skip_op = std::make_shared(5); rc = my_tree->AssociateNode(skip_op); ASSERT_TRUE(rc.IsOk()); // Set children/root layout. rc = skip_op->AddChild(my_tfreader_op); ASSERT_TRUE(rc.IsOk()); rc = my_tree->AssignRoot(skip_op); ASSERT_TRUE(rc.IsOk()); MS_LOG(INFO) << "Launching tree and begin iteration."; rc = my_tree->Prepare(); ASSERT_TRUE(rc.IsOk()); rc = my_tree->Launch(); ASSERT_TRUE(rc.IsOk()); // Start the loop of reading tensors from our pipeline DatasetIterator di(my_tree); TensorRow tensor_list; rc = di.FetchNextTensorRow(&tensor_list); ASSERT_TRUE(rc.IsOk()); int row_count = 0; while (!tensor_list.empty()) { MS_LOG(INFO) << "Row display for row #: " << row_count << "."; // Display the tensor by calling the printer on it for (int i = 0; i < tensor_list.size(); i++) { std::ostringstream ss; ss << "(" << tensor_list[i] << "): " << *tensor_list[i] << std::endl; MS_LOG(INFO) << "Tensor print: " << ss.str() << "."; } rc = di.FetchNextTensorRow(&tensor_list); ASSERT_TRUE(rc.IsOk()); row_count++; } ASSERT_EQ(row_count, 7); }