From e32ab2975c25e7533bcb9e1f8ff4b0dbaec83476 Mon Sep 17 00:00:00 2001
From: jwx930962 <jiangcheng22@hisilicon.com>
Date: Thu, 16 Dec 2021 14:47:20 +0800
Subject: [PATCH] increase st testcase

---
 .../st/testcase/origin_models/caffe_add.caffemodel | Bin 0 -> 153 bytes
 tests/st/testcase/origin_models/test.json          |  71 ------------
 tests/st/testcase/test_caffe_parser.cc             | 121 +++++++++++++++++++++
 3 files changed, 121 insertions(+), 71 deletions(-)
 create mode 100644 tests/st/testcase/origin_models/caffe_add.caffemodel
 delete mode 100644 tests/st/testcase/origin_models/test.json

diff --git a/tests/st/testcase/origin_models/caffe_add.caffemodel b/tests/st/testcase/origin_models/caffe_add.caffemodel
new file mode 100644
index 0000000000000000000000000000000000000000..34efbb386789f3e6f0ab4852965e6ae05db627e9
GIT binary patch
literal 153
zcmV;K0A~LR2UKNqbU|cf8U$ovbYY?f8wvzuVRT^<1xapjb#x*GWMOn+`UnLI0}25H
zq6RVw1!He+b952~Np5g;bRq>~Z*FsRIuj}k004kL0000$002NhIs*y;1NsOB3IhrO
z1EK~r3I}p!b7*03WfBKcWpij@aAg_<WMOn+8U<r-ZgX@Z3vy+1XmDX=UvG7Est5%N
H0}25H$~`A7

literal 0
KcmV+b0RR6000031

diff --git a/tests/st/testcase/origin_models/test.json b/tests/st/testcase/origin_models/test.json
deleted file mode 100644
index d1dbcc4..0000000
--- a/tests/st/testcase/origin_models/test.json
+++ /dev/null
@@ -1,71 +0,0 @@
-{
-  "node": [
-    {
-      "attr": [
-        {
-          "key": "dtype",
-          "value": {
-            "type": "DT_HALF"
-          }
-        },
-        {
-          "key": "shape",
-          "value": {
-            "shape": {
-              "dim": [
-                {
-                  "size": 1
-                }
-              ]
-            }
-          }
-        }
-      ],
-      "name": "Placeholder",
-      "op": "Placeholder"
-    },
-    {
-      "attr": [
-        {
-          "key": "dtype",
-          "value": {
-            "type": "DT_HALF"
-          }
-        },
-        {
-          "key": "shape",
-          "value": {
-            "shape": {
-              "dim": [
-                {
-                  "size": 1
-                }
-              ]
-            }
-          }
-        }
-      ],
-      "name": "Placeholder_1",
-      "op": "Placeholder"
-    },
-    {
-      "attr": [
-        {
-          "key": "T",
-          "value": {
-            "type": "DT_HALF"
-          }
-        }
-      ],
-      "input": [
-        "Placeholder",
-        "Placeholder_1"
-      ],
-      "name": "add_test_1",
-      "op": "Add"
-    }
-  ],
-  "versions": {
-    "producer": 134
-  }
-}
\ No newline at end of file
diff --git a/tests/st/testcase/test_caffe_parser.cc b/tests/st/testcase/test_caffe_parser.cc
index dc947e5..3f8dbb3 100644
--- a/tests/st/testcase/test_caffe_parser.cc
+++ b/tests/st/testcase/test_caffe_parser.cc
@@ -181,6 +181,14 @@ TEST_F(STestCaffeParser, caffe_parser_ParseParamsForDummyData_test)
 
   ret = caffe_parser.ParseParamsForInput(lay, op);
   EXPECT_EQ(ret, FAILED);
+
+  domi::caffe::DummyDataParameter *dummyData = lay->mutable_dummy_data_param();
+  ret = caffe_parser.ParseParamsForDummyData(lay, op);
+  EXPECT_EQ(ret, FAILED);
+
+  domi::caffe::BlobShape* dummpShape = dummyData->add_shape();
+  ret = caffe_parser.ParseParamsForDummyData(lay, op);
+  EXPECT_EQ(ret, SUCCESS);
 }
 
 TEST_F(STestCaffeParser, convertWeights_success)
@@ -363,4 +371,117 @@ TEST_F(STestCaffeParser, CaffeWeightsParser_ParseOutputNodeTopInfo_test)
   EXPECT_EQ(ret, PARAM_INVALID);
 }
 
+TEST_F(STestCaffeParser, CaffeOpParser_ParseWeightType_test)
+{
+  CaffeOpParser opParser;
+  ge::GeTensorDesc ge_tensor_desc =  ge::GeTensorDesc();
+  ge::GeTensorPtr weight = std::make_shared<ge::GeTensor>(ge_tensor_desc);
+  ge::OpDescPtr opDef = std::make_shared<ge::OpDesc>("","");
+  auto node_tmp = GenNodeFromOpDesc(opDef);
+
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  blob->set_int8_data("10");
+  std::string lay_name = "DATA";
+  GeShape shape({1,1,3,4});
+  Status ret = opParser.ParseWeightType(*blob, shape, 1, lay_name, weight);
+  EXPECT_EQ(ret, FAILED);
+}
+
+TEST_F(STestCaffeParser, CaffeOpParser_ParseWeightType_test2)
+{
+  CaffeOpParser opParser;
+  ge::GeTensorDesc ge_tensor_desc =  ge::GeTensorDesc();
+  ge::GeTensorPtr weight = std::make_shared<ge::GeTensor>(ge_tensor_desc);
+  ge::OpDescPtr opDef = std::make_shared<ge::OpDesc>("","");
+  auto node_tmp = GenNodeFromOpDesc(opDef);
+
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  blob->add_int32_data(10);
+
+  std::string lay_name = "DATA";
+  GeShape shape({1,1,3,4});
+  Status ret = opParser.ParseWeightType(*blob, shape, 1, lay_name, weight);
+  EXPECT_EQ(ret, SUCCESS);
+
+  ret = opParser.ParseWeightType(*blob, shape, 2, lay_name, weight);
+  EXPECT_EQ(ret, FAILED);
+}
+
+TEST_F(STestCaffeParser, CaffeOpParser_ParseWeightType_test3)
+{
+  CaffeOpParser opParser;
+  ge::GeTensorDesc ge_tensor_desc =  ge::GeTensorDesc();
+  ge::GeTensorPtr weight = std::make_shared<ge::GeTensor>(ge_tensor_desc);
+  ge::OpDescPtr opDef = std::make_shared<ge::OpDesc>("","");
+  auto node_tmp = GenNodeFromOpDesc(opDef);
+
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  double value = 2.0;
+  blob->add_double_data(value);
+
+  std::string lay_name = "DATA";
+  GeShape shape({1,1,3,4});
+  Status ret = opParser.ParseWeightType(*blob, shape, 1, lay_name, weight);
+  EXPECT_EQ(ret, SUCCESS);
+
+  ret = opParser.ParseWeightType(*blob, shape, 3, lay_name, weight);
+  EXPECT_EQ(ret, FAILED);
+}
+
+TEST_F(STestCaffeParser, CaffeOpParser_ParseWeightType_test4)
+{
+  CaffeOpParser opParser;
+  ge::GeTensorDesc ge_tensor_desc =  ge::GeTensorDesc();
+  ge::GeTensorPtr weight = std::make_shared<ge::GeTensor>(ge_tensor_desc);
+  ge::OpDescPtr opDef = std::make_shared<ge::OpDesc>("","");
+  auto node_tmp = GenNodeFromOpDesc(opDef);
+
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  blob->add_uint64_data(10);
+
+  std::string lay_name = "DATA";
+  GeShape shape({1,1,3,4});
+  Status ret = opParser.ParseWeightType(*blob, shape, 1, lay_name, weight);
+  EXPECT_EQ(ret, SUCCESS);
+
+  ret = opParser.ParseWeightType(*blob, shape, 2, lay_name, weight);
+  EXPECT_EQ(ret, FAILED);
+}
+
+TEST_F(STestCaffeParser, CaffeOpParser_ParseWeightType_test5)
+{
+  CaffeOpParser opParser;
+  ge::GeTensorDesc ge_tensor_desc =  ge::GeTensorDesc();
+  ge::GeTensorPtr weight = std::make_shared<ge::GeTensor>(ge_tensor_desc);
+  ge::OpDescPtr opDef = std::make_shared<ge::OpDesc>("","");
+  auto node_tmp = GenNodeFromOpDesc(opDef);
+
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  blob->add_data(10);
+
+  std::string lay_name = "DATA";
+  GeShape shape({1,1,3,4});
+  Status ret = opParser.ParseWeightType(*blob, shape, 10, lay_name, weight);
+  EXPECT_EQ(ret, FAILED);
+}
+
+TEST_F(STestCaffeParser, CaffeOpParser_ConvertShape_test)
+{
+  CaffeOpParser opParser;
+  domi::caffe::LayerParameter *layer = new domi::caffe::LayerParameter();
+  domi::caffe::BlobProto *blob = layer->add_blobs();
+  blob->set_num(1);
+  blob->set_channels(2);
+  blob->set_height(1);
+  blob->set_width(1);
+  std::vector<int64_t> shape;
+
+  opParser.ConvertShape(*blob, shape);
+}
+
 } // namespace ge