From 0646b4b6efd78c87b19ed9fff4683d32bc1ab767 Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Thu, 4 Mar 2021 15:32:42 +0800 Subject: [PATCH] modified: tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc --- .../graph/passes/replace_with_empty_const_pass_unittest.cc | 6 ++++++ 1 file changed, 6 insertions(+) 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);