Add unit test for no NetOutput

4 years ago · 75bcf24692
parent dd9099f661
commit 75bcf24692
1 changed files with 1 additions and 1 deletions
--- a/tests/ut/ge/graph/load/davinci_model_unittest.cc
+++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc
@ -345,7 +345,7 @@ TEST_F(UtestDavinciModel, Init_variable_op) {

  EXPECT_EQ(model.InitNodes(graph), SUCCESS);

-  EXPECT_EQ(model.ReturnNoOutput(1), PARAM_INVALID);
+  EXPECT_EQ(model.ReturnNoOutput(1), SUCCESS);
  EXPECT_EQ(model.SyncVarData(), SUCCESS);

  OutputData output_data;