From 74d0a7da19f45eeeb871eb855c359f3c53bfd0f2 Mon Sep 17 00:00:00 2001
From: 13291271729 <xuyafei3@huawei.com>
Date: Wed, 31 Aug 2022 11:35:23 +0800
Subject: [PATCH] clean code

---
 .../tensorflow_parser_unittest.cc             | 50 +++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc
index 00678d8..5f2ee69 100644
--- a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc
+++ b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc
@@ -3622,6 +3622,56 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_tes
   EXPECT_EQ(ret, SUCCESS);
 }
 
+TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_success)
+{
+  Message2Operator mess2Op;
+  tensorflow::NodeDef nodedef;
+  nodedef.set_name(_name);
+  nodedef.set_op(_name);
+
+  ge::OpDescPtr op_desc = std::make_shared<ge::OpDesc>();
+  ge::Operator ops = ge::OpDescUtils::CreateOperatorFromOpDesc(op_desc);
+
+  tensorflow::AttrValue value;
+  value.set_s("string");
+  TensorFlowUtil::AddNodeAttr("op_def", value, &nodedef);
+  value.set_i(1);
+  TensorFlowUtil::AddNodeAttr("num", value, &nodedef);
+  domi::tensorflow::DataType VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT;
+  value.set_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("float", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_UINT32;
+  value.set_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("uint32", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_INT64;
+  value.set_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("int64", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL;
+  value.set_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("bool", value, &nodedef);
+  Status ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops);
+  EXPECT_EQ(ret, SUCCESS);
+
+  value.mutable_list()->set_s("string");
+  TensorFlowUtil::AddNodeAttr("op_def", value, &nodedef);
+  value.mutable_list()->set_i(1);
+  TensorFlowUtil::AddNodeAttr("num", value, &nodedef);
+  domi::tensorflow::DataType VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT;
+  value.mutable_list()->add_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("float", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_UINT32;
+  value.mutable_list()->add_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("uint32", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_INT64;
+  value.mutable_list()->add_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("int64", value, &nodedef);
+  VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL;
+  value.mutable_list()->add_type(VALUE_TYPE);
+  TensorFlowUtil::AddNodeAttr("bool", value, &nodedef);
+  ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops);
+  EXPECT_EQ(ret, SUCCESS);
+}
+
 TEST_F(UtestTensorflowParser, tensorflow_Pb2Json_RepeatedEnum2Json_test)
 {
   Pb2Json toJson;