clean code

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;