diff --git a/tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc b/tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc index 348543d3..d7984fb3 100644 --- a/tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc +++ b/tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc @@ -74,6 +74,12 @@ TEST_F(UtestReplaceWithEmptyConstPass, replace_whith_empty_const_success) { EXPECT_EQ(ret, SUCCESS); EXPECT_EQ(graph->GetDirectNodesSize(),5); + // run pass on const1, graph still has 5 nodes + auto const1 = graph->FindNode("const1"); + Status ret = replace_with_empty_const_pass.Run(const1); + EXPECT_EQ(ret, SUCCESS); + EXPECT_EQ(graph->GetDirectNodesSize(),5); + auto cast1 = graph->FindNode("cast1"); ret = replace_with_empty_const_pass.Run(cast1); EXPECT_EQ(cast1->GetOutAllNodes().size(),0);