diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc
index eda3cb15..a9efab3d 100644
--- 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;
 }