From 2ac7fab49b135d049acbcd19c00b6ed54ca565cf Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Mon, 4 Jan 2021 15:25:11 +0800 Subject: [PATCH] Modify UT for GetAippType --- tests/ut/ge/graph/load/davinci_model_unittest.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc index fa596d36..d5b706d7 100644 --- a/tests/ut/ge/graph/load/davinci_model_unittest.cc +++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc @@ -385,7 +385,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_static) { InputAippType aipp_type; size_t aipp_index = 0; - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.InitNodes(graph), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(aipp_type, DATA_WITH_STATIC_AIPP); @@ -417,9 +417,9 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic) { InputAippType aipp_type; size_t aipp_index = 0; - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.InitNodes(graph), SUCCESS); - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.input_addrs_list_.size(), 1); EXPECT_EQ(model.output_addrs_list_.size(), 0); @@ -457,7 +457,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_releated) { InputAippType aipp_type; size_t aipp_index = 0; - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.InitNodes(graph), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(aipp_type, DATA_WITH_DYNAMIC_AIPP); @@ -488,7 +488,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic_conf) { InputAippType aipp_type; size_t aipp_index = 0; - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.InitNodes(graph), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(aipp_type, DYNAMIC_AIPP_NODE); @@ -519,7 +519,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic_invalid) { InputAippType aipp_type; size_t aipp_index = 0; - EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), ACL_ERROR_GE_AIPP_NOT_EXIST); + EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.InitNodes(graph), ACL_ERROR_GE_AIPP_MODE_INVALID); EXPECT_EQ(model.input_addrs_list_.size(), 1);