!2374 C++ UT for CutOut and Pad

Merge pull request !2374 from MahdiRahmaniHanzaki/I1L892-pad-cutout-ut

tests/ut/cpp/dataset/CMakeLists.txt

@@ -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

tests/ut/cpp/dataset/cut_out_op_test.cc

+/**
+ * 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());
+}
+

tests/ut/cpp/dataset/pad_op_test.cc

+/**
+ * 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());
+}