diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc index 3cd0455d..34d45269 100644 --- a/tests/ut/ge/graph/load/davinci_model_unittest.cc +++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc @@ -282,4 +282,42 @@ TEST_F(UtestDavinciModel, init_unknown) { const vector outputs = { &virtual_addr }; EXPECT_EQ(model.UpdateKnownNodeArgs(inputs, outputs), SUCCESS); } + +TEST_F(UtestDavinciModel, ReturnNoOutput_test) { + DavinciModel model(0, nullptr); + + GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); + TensorUtils::SetSize(tensor, 512); + + OpDescPtr var1 = CreateOpDesc("var1", VARIABLE); + var1->AddInputDesc(tensor); + var1->AddOutputDesc(tensor); + var1->SetInputOffset({1024}); + var1->SetOutputOffset({1024}); + + model.variable_op_list_.push_back(var1); + + + EXPECT_EQ(model.ReturnNoOutput(model), PARAM_INVALID); +} + +TEST_F(UtestDavinciModel, SyncVarData_test) { + DavinciModel model(0, nullptr); + + GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); + TensorUtils::SetSize(tensor, 512); + + OpDescPtr var1 = CreateOpDesc("var1", VARIABLE); + var1->AddInputDesc(tensor); + var1->AddOutputDesc(tensor); + var1->SetInputOffset({1024}); + var1->SetOutputOffset({1024}); + + model.variable_op_list_.push_back(var1); + + EXPECT_NE(model.SyncVarData(model), SUCCESS); + +} + + } // namespace ge