Huawei_Technology
/
MindSpore-GraphEngine
Not watched
1
0
Fork 0
Code
Releases 10 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

Modify UT for GetAippType

pull/846/head
zhangxiaokun 5 years ago
parent
ec90d7afeb
commit
2ac7fab49b
1 changed files with 6 additions and 6 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +6
    -6
      tests/ut/ge/graph/load/davinci_model_unittest.cc

+ 6
- 6
tests/ut/ge/graph/load/davinci_model_unittest.cc View File

@@ -385,7 +385,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_static) {


InputAippType aipp_type; InputAippType aipp_type;
size_t aipp_index = 0; 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.InitNodes(graph), SUCCESS);
EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS);
EXPECT_EQ(aipp_type, DATA_WITH_STATIC_AIPP); EXPECT_EQ(aipp_type, DATA_WITH_STATIC_AIPP);
@@ -417,9 +417,9 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic) {


InputAippType aipp_type; InputAippType aipp_type;
size_t aipp_index = 0; 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.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.input_addrs_list_.size(), 1);
EXPECT_EQ(model.output_addrs_list_.size(), 0); EXPECT_EQ(model.output_addrs_list_.size(), 0);
@@ -457,7 +457,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_releated) {


InputAippType aipp_type; InputAippType aipp_type;
size_t aipp_index = 0; 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.InitNodes(graph), SUCCESS);
EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS);
EXPECT_EQ(aipp_type, DATA_WITH_DYNAMIC_AIPP); EXPECT_EQ(aipp_type, DATA_WITH_DYNAMIC_AIPP);
@@ -488,7 +488,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic_conf) {


InputAippType aipp_type; InputAippType aipp_type;
size_t aipp_index = 0; 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.InitNodes(graph), SUCCESS);
EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS); EXPECT_EQ(model.GetAippType(0, aipp_type, aipp_index), SUCCESS);
EXPECT_EQ(aipp_type, DYNAMIC_AIPP_NODE); EXPECT_EQ(aipp_type, DYNAMIC_AIPP_NODE);
@@ -519,7 +519,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_dynamic_invalid) {


InputAippType aipp_type; InputAippType aipp_type;
size_t aipp_index = 0; 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.InitNodes(graph), ACL_ERROR_GE_AIPP_MODE_INVALID);


EXPECT_EQ(model.input_addrs_list_.size(), 1); EXPECT_EQ(model.input_addrs_list_.size(), 1);


Write Preview
Loading…
Cancel
Save