Browse Source

Add UT

tags/v1.2.0
zhangxiaokun 4 years ago
parent
commit
5bedbf9696
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      tests/ut/ge/graph/load/davinci_model_unittest.cc

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

@@ -111,6 +111,12 @@ TEST_F(UtestDavinciModel, init_success) {
EXPECT_EQ(model.output_addrs_list_.size(), 1);
EXPECT_EQ(model.task_list_.size(), 2);

OutputData output_data;
vector<OutputTensorInfo> outputs;
EXPECT_EQ(model.GenOutputTensorInfo(&output_data, outputs), SUCCESS);
EXPECT_EQ(output_data.blobs.size(), 1);
EXPECT_EQ(outputs.size(), 1);

ProfilingManager::Instance().is_load_profiling_ = false;
}



Loading…
Cancel
Save