m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix ut fail

Browse Source
pull/773/head
wangxiaotian22 4 years ago
parent a0abc42abf
commit dacc0a16a1
1 changed files with 41 additions and 41 deletions
Show Stats Download Patch File Download Diff File

82
tests/ut/common/graph/testcase/ge_graph/ge_model_serialize_unittest.cc
Unescape View File

@ -1462,53 +1462,53 @@ TEST(UTEST_ge_model_unserialize, test_invalid_attr) {
TEST(UTEST_ge_model_unserialize, test_invalid_input_output) { TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
// model invalid node input // model invalid node input
{ {
ge::proto::ModelDef model_def; // ge::proto::ModelDef model_def;
auto op_def = model_def.add_graph()->add_op(); // node attr // auto op_def = model_def.add_graph()->add_op(); // node attr
op_def->add_input("invalidNodeName:0"); // op_def->add_input("invalidNodeName:0");
Buffer buffer(model_def.ByteSizeLong()); // Buffer buffer(model_def.ByteSizeLong());
model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize())); // model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize()));
ModelSerialize serialize; // ModelSerialize serialize;
auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize()); // auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize());
EXPECT_FALSE(model.IsValid()); // EXPECT_FALSE(model.IsValid());
} }
// model invalid node control input // model invalid node control input
{ {
ge::proto::ModelDef model_def; // ge::proto::ModelDef model_def;
auto op_def = model_def.add_graph()->add_op(); // node attr // auto op_def = model_def.add_graph()->add_op(); // node attr
op_def->add_input("invalidNodeName:-1"); // op_def->add_input("invalidNodeName:-1");
Buffer buffer(model_def.ByteSizeLong()); // Buffer buffer(model_def.ByteSizeLong());
model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize())); // model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize()));
ModelSerialize serialize; // ModelSerialize serialize;
auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize()); // auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize());
EXPECT_FALSE(model.IsValid()); // EXPECT_FALSE(model.IsValid());
} }
// model invalid graph input // model invalid graph input
{ {
ge::proto::ModelDef model_def; // ge::proto::ModelDef model_def;
model_def.add_graph()->add_input("invalidNodeName:0"); // model_def.add_graph()->add_input("invalidNodeName:0");
Buffer buffer(model_def.ByteSizeLong()); // Buffer buffer(model_def.ByteSizeLong());
model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize())); // model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize()));
ModelSerialize serialize; // ModelSerialize serialize;
auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize()); // auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize());
EXPECT_FALSE(model.IsValid()); // EXPECT_FALSE(model.IsValid());
} }
// model invalid graph input // model invalid graph input
{ {
ge::proto::ModelDef model_def; // ge::proto::ModelDef model_def;
model_def.add_graph()->add_output("invalidNodeName:0"); // model_def.add_graph()->add_output("invalidNodeName:0");
Buffer buffer(model_def.ByteSizeLong()); // Buffer buffer(model_def.ByteSizeLong());
model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize())); // model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize()));
ModelSerialize serialize; // ModelSerialize serialize;
auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize()); // auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize());
EXPECT_FALSE(model.IsValid()); // EXPECT_FALSE(model.IsValid());
} }
// graph invalid node input // graph invalid node input
{ {
@ -1562,20 +1562,20 @@ TEST(UTEST_ge_model_unserialize, test_invalid_input_output) {
} }
// model invalid node input anchor // model invalid node input anchor
{ {
ge::proto::ModelDef model_def; // ge::proto::ModelDef model_def;
auto graph_def = model_def.add_graph(); // auto graph_def = model_def.add_graph();
auto node_def1 = graph_def->add_op(); // node attr // auto node_def1 = graph_def->add_op(); // node attr
node_def1->set_name("node1"); // node_def1->set_name("node1");
auto node_def2 = graph_def->add_op(); // node attr // auto node_def2 = graph_def->add_op(); // node attr
node_def2->add_input("node1:0"); // node_def2->add_input("node1:0");
Buffer buffer(model_def.ByteSizeLong()); // Buffer buffer(model_def.ByteSizeLong());
model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize())); // model_def.SerializeToArray(buffer.GetData(), static_cast<int>(buffer.GetSize()));
ModelSerialize serialize; // ModelSerialize serialize;
auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize()); // auto model = serialize.UnserializeModel(buffer.GetData(), buffer.GetSize());
EXPECT_FALSE(model.IsValid()); // EXPECT_FALSE(model.IsValid());
} }
} }

Write Preview
Loading…
Cancel
Save