modified: tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc

4 years ago · 0646b4b6ef
parent 554d08f277
commit 0646b4b6ef
1 changed files with 6 additions and 0 deletions
--- 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);