From 388ba4b7d8e0007da90243fdc07bc7d563f1770e Mon Sep 17 00:00:00 2001 From: 13291271729 Date: Wed, 31 Aug 2022 14:21:40 +0800 Subject: [PATCH] clean code --- .../tensorflow_parser_unittest.cc | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) 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 3a6c6e4..2119f78 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 @@ -3654,14 +3654,15 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc 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); + Status ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops); EXPECT_EQ(ret, SUCCESS); value.clear_value(); - value.mutable_list()->set_s("string"); + VALUE_TYPE = domi::tensorflow::DataType::DT_STRING; + value.mutable_list()->add_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("str_list", value, &nodedef); value.clear_value(); - value.mutable_list()->set_i(1); + value.mutable_list()->add_i(1); TensorFlowUtil::AddNodeAttr("num_list", value, &nodedef); value.clear_value(); VALUE_TYPE = domi::tensorflow::DataType::DT_FLOAT; @@ -3679,7 +3680,7 @@ TEST_F(UtestTensorflowParser, tensorflow_Message2Operator_ParseOperatorAttrs_suc VALUE_TYPE = domi::tensorflow::DataType::DT_BOOL; value.mutable_list()->add_type(VALUE_TYPE); TensorFlowUtil::AddNodeAttr("bool_list", value, &nodedef); - ret = mess2Op.ParseOperatorAttrs(&node_def, 1, ops); + ret = mess2Op.ParseOperatorAttrs(&nodedef, 1, ops); EXPECT_EQ(ret, SUCCESS); }