| @@ -17,6 +17,7 @@ SET(DE_UT_SRCS | |||
| circular_pool_test.cc | |||
| client_config_test.cc | |||
| connector_test.cc | |||
| cut_out_op_test.cc | |||
| datatype_test.cc | |||
| decode_op_test.cc | |||
| execution_tree_test.cc | |||
| @@ -28,6 +29,7 @@ SET(DE_UT_SRCS | |||
| normalize_op_test.cc | |||
| one_hot_op_test.cc | |||
| pad_end_op_test.cc | |||
| pad_op_test.cc | |||
| path_test.cc | |||
| project_op_test.cc | |||
| queue_test.cc | |||
| @@ -0,0 +1,44 @@ | |||
| /** | |||
| * 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 "common/common.h" | |||
| #include "common/cvop_common.h" | |||
| #include "dataset/kernels/image/cut_out_op.h" | |||
| #include "utils/log_adapter.h" | |||
| using namespace mindspore::dataset; | |||
| using mindspore::MsLogLevel::INFO; | |||
| using mindspore::ExceptionType::NoExceptionType; | |||
| using mindspore::LogStream; | |||
| class MindDataTestCutOutOp : public UT::CVOP::CVOpCommon { | |||
| protected: | |||
| MindDataTestCutOutOp() : CVOpCommon() {} | |||
| std::shared_ptr<Tensor> output_tensor_; | |||
| }; | |||
| TEST_F(MindDataTestCutOutOp, TestOp) { | |||
| MS_LOG(INFO) << "Doing testCutOut."; | |||
| std::unique_ptr<CutOutOp> op(new CutOutOp(50, 50, 5, false, 0, 0, 0)); | |||
| EXPECT_TRUE(op->OneToOne()); | |||
| Status s = op->Compute(input_tensor_, &output_tensor_); | |||
| EXPECT_EQ(input_tensor_->shape()[0], output_tensor_->shape()[0]); | |||
| EXPECT_EQ(input_tensor_->shape()[1], output_tensor_->shape()[1]); | |||
| EXPECT_EQ(input_tensor_->shape()[2], output_tensor_->shape()[2]); | |||
| EXPECT_EQ(s, Status::OK()); | |||
| } | |||
| @@ -0,0 +1,44 @@ | |||
| /** | |||
| * 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 "common/common.h" | |||
| #include "common/cvop_common.h" | |||
| #include "dataset/kernels/image/pad_op.h" | |||
| #include "utils/log_adapter.h" | |||
| using namespace mindspore::dataset; | |||
| using mindspore::MsLogLevel::INFO; | |||
| using mindspore::ExceptionType::NoExceptionType; | |||
| using mindspore::LogStream; | |||
| class MindDataTestPadOp : public UT::CVOP::CVOpCommon { | |||
| protected: | |||
| MindDataTestPadOp() : CVOpCommon() {} | |||
| std::shared_ptr<Tensor> output_tensor_; | |||
| }; | |||
| TEST_F(MindDataTestPadOp, TestOp) { | |||
| MS_LOG(INFO) << "Doing testPad."; | |||
| std::unique_ptr<PadOp> op(new PadOp(10, 20, 30, 40, BorderType::kConstant)); | |||
| EXPECT_TRUE(op->OneToOne()); | |||
| Status s = op->Compute(input_tensor_, &output_tensor_); | |||
| size_t actual = 0; | |||
| if (s == Status::OK()) { | |||
| actual = output_tensor_->shape()[0] * output_tensor_->shape()[1] * output_tensor_->shape()[2]; | |||
| } | |||
| EXPECT_EQ(actual, (input_tensor_->shape()[0] + 30) * (input_tensor_->shape()[1] + 70) * 3); | |||
| EXPECT_EQ(s, Status::OK()); | |||
| } | |||