Browse Source

fill ut

tags/v1.2.0
wangxiaotian22 4 years ago
parent
commit
610828561c
1 changed files with 38 additions and 0 deletions
  1. +38
    -0
      tests/ut/ge/graph/load/davinci_model_unittest.cc

+ 38
- 0
tests/ut/ge/graph/load/davinci_model_unittest.cc View File

@@ -282,4 +282,42 @@ TEST_F(UtestDavinciModel, init_unknown) {
const vector<void *> 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

Loading…
Cancel
Save