modified: tests/ut/ge/graph/load/new_model_manager_model_manager_unittest.cc

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