From ac9dda90b7a1b507e6bf8acb2f071e32bf1b4dca Mon Sep 17 00:00:00 2001 From: jwx930962 Date: Sat, 18 Dec 2021 17:51:27 +0800 Subject: [PATCH] increase ut testcase --- .../tensorflow_parser_unittest.cc | 6 ++++++ 1 file changed, 6 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 3ff888a..fa7c547 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 @@ -2657,6 +2657,8 @@ TEST_F(UtestTensorflowParser, Tensorflow_UpdateFusionOpContext_test) Status ret = tensorflow_parser.UpdateFusionOpContext(scope_graph, info, fusion_op_node_context, normal_op_node_context); EXPECT_EQ(ret, domi::SUCCESS); + + delete graph; } @@ -2728,6 +2730,10 @@ TEST_F(UtestTensorflowParser, Tensorflow_GetInOutPutIndex_scope_pass) ret = tensorflow_parser.GetOutPutIndex(scope_graph, output_node_info, old_index, new_index); EXPECT_EQ(domi::SUCCESS, ret); EXPECT_EQ(true, (new_index == 1)); + + input_node_info.scope_pass = false; + ret = tensorflow_parser.GetInPutIndex(scope_graph, input_node_info, old_index, new_index); + EXPECT_EQ(domi::SUCCESS, ret); delete graph; }