From 2782e71a11082edf83906efaecfc5c0a47731a72 Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Sun, 22 Jul 2018 12:43:27 +0800 Subject: [PATCH] fix graph_test --- paddle/fluid/framework/ir/graph_test.cc | 26 ++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/paddle/fluid/framework/ir/graph_test.cc b/paddle/fluid/framework/ir/graph_test.cc index feb4e8e76b..73ef55756c 100644 --- a/paddle/fluid/framework/ir/graph_test.cc +++ b/paddle/fluid/framework/ir/graph_test.cc @@ -95,19 +95,19 @@ TEST(GraphTest, Basic) { std::unique_ptr g(new ir::Graph(prog)); std::vector nodes(g->Nodes().begin(), g->Nodes().end()); - ASSERT_EQ(nodes[0]->Name(), "sum"); - ASSERT_EQ(nodes[0]->inputs[0]->Name(), "test_a"); - ASSERT_EQ(nodes[0]->inputs[1]->Name(), "test_b"); - ASSERT_EQ(nodes[0]->inputs[2]->Name(), "test_c"); - ASSERT_EQ(nodes[0]->outputs[0]->Name(), "test_out"); - ASSERT_EQ(nodes[1]->Name(), "test_a"); - ASSERT_EQ(nodes[1]->outputs[0]->Name(), "sum"); - ASSERT_EQ(nodes[2]->Name(), "test_b"); - ASSERT_EQ(nodes[2]->outputs[0]->Name(), "sum"); - ASSERT_EQ(nodes[3]->Name(), "test_c"); - ASSERT_EQ(nodes[3]->outputs[0]->Name(), "sum"); - ASSERT_EQ(nodes[4]->Name(), "test_out"); - ASSERT_EQ(nodes[4]->inputs[0]->Name(), "sum"); + for (ir::Node *n : nodes) { + if (n->Name() == "sum") { + ASSERT_EQ(n->inputs.size(), 3); + ASSERT_EQ(n->outputs.size(), 1); + } else if (n->Name() == "test_a" || n->Name() == "test_b" || + n->Name() == "test_c") { + ASSERT_EQ(n->inputs.size(), 0); + ASSERT_EQ(n->outputs.size(), 1); + } else if (n->Name() == "test_out") { + ASSERT_EQ(n->inputs.size(), 1); + ASSERT_EQ(n->outputs.size(), 0); + } + } ASSERT_EQ(nodes.size(), 5); } } // namespace framework -- GitLab