Add UT

4 years ago · 5bedbf9696
parent 0c14832647
commit 5bedbf9696
1 changed files with 6 additions and 0 deletions
--- a/tests/ut/ge/graph/load/davinci_model_unittest.cc
+++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc
@ -111,6 +111,12 @@ TEST_F(UtestDavinciModel, init_success) {
  EXPECT_EQ(model.output_addrs_list_.size(), 1);
  EXPECT_EQ(model.task_list_.size(), 2);

+  OutputData output_data;
+  vector<OutputTensorInfo> outputs;
+  EXPECT_EQ(model.GenOutputTensorInfo(&output_data, outputs), SUCCESS);
+  EXPECT_EQ(output_data.blobs.size(), 1);
+  EXPECT_EQ(outputs.size(), 1);
+
  ProfilingManager::Instance().is_load_profiling_ = false;
 }