From 2b174a623141289db53e2d244c4297f369f9a9df Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Tue, 19 Jan 2021 15:35:13 +0800
Subject: [PATCH] 	modified:  
 tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc

---
 .../new_model_manager_model_manager_unittest.cc     | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc b/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc
index 1c6e5a10..4096afa5 100644
--- a/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc
+++ b/tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc
@@ -348,5 +348,18 @@ TEST_F(UtestModelManagerModelManager, destroy_aicpu_session) {
   manager.sess_ids_.insert(0);
   manager.DestroyAicpuSession(0);
 }
+// test DataInputTensor
+TEST_F(UtestModelManagerModelManager, test_data_input_tensor) {
+  ModelManager mm;
+  mm.model_map_={{1,make_shared<DavinciModel>()}};
+  mm.hybrid_model_map_={{1,make_shared<ybrid::HybridDavinciModel>()}};
+  ge::ModelData data;
+  data.model_len = 10;
+  uint32_t model_id = 1;
+  InputTensorInfo input_tensor = new InputTensorInfo();
+  std::vector<InputTensorInfo> inputs = {input_tensor};
+  auto ret = mm.DataInputTensor(1,inputs);
+  EXPECT_EQ(ge::SUCCESS, ret);
+}
 
 }  // namespace ge