From 312abfca621f14008a0f9a1a751c2f9f3c300133 Mon Sep 17 00:00:00 2001
From: yuxing <yuxing11@hisilicon.com>
Date: Fri, 6 Aug 2021 11:29:10 +0800
Subject: [PATCH] modify error_code related by op

---
 tests/ut/ge/generator/ge_generator_unittest.cc | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc
index a4f2f6d7..8e24919c 100644
--- a/tests/ut/ge/generator/ge_generator_unittest.cc
+++ b/tests/ut/ge/generator/ge_generator_unittest.cc
@@ -188,9 +188,12 @@ TEST_F(UtestGeGenerator, test_generate_online_model) {
 }
 
 TEST_F(UtestGeGenerator, test_check_engine_type_support) {
+  GeTensorDesc tensor_desc;
   shared_ptr<OpDesc> op_desc = make_shared<OpDesc>("Add", "add");
-  GeGenerator generator;
-  generator.Initialize({});
-  EXPECT_EQ(generator.CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED);
+  op_desc->AddInputDesc(tensor_desc);
+  op_desc->AddInputDesc(tensor_desc);
+  op_desc->AddOutputDesc(tensor_desc);
+  
+  EXPECT_EQ(CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED);
 }
 }  // namespace ge