From 5bedbf96964f25e47c29eaf2e7d24495dd05ea95 Mon Sep 17 00:00:00 2001
From: zhangxiaokun <zhang.xiaokun@huawei.com>
Date: Thu, 31 Dec 2020 15:30:25 +0800
Subject: [PATCH] Add UT

---
 tests/ut/ge/graph/load/davinci_model_unittest.cc | 6 ++++++
 1 file changed, 6 insertions(+)

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;
 }