Browse Source

ut

tags/v1.3.0
wjm 3 years ago
parent
commit
37e4568ce3
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 2
- 0
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

@@ -157,6 +157,7 @@ TEST_F(UtestGeHybrid, index_taskdefs_failed) {
ge_root_model->SetModelName("test_name"); ge_root_model->SetModelName("test_name");
HybridModel hybrid_model(ge_root_model); HybridModel hybrid_model(ge_root_model);
HybridModelBuilder hybrid_model_builder(hybrid_model); HybridModelBuilder hybrid_model_builder(hybrid_model);

ASSERT_EQ(hybrid_model_builder.Build(), INTERNAL_ERROR); ASSERT_EQ(hybrid_model_builder.Build(), INTERNAL_ERROR);
ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), INTERNAL_ERROR); ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), INTERNAL_ERROR);
} }
@@ -241,6 +242,7 @@ TEST_F(UtestGeHybrid, index_taskdefs_success) {
GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph); GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph);
HybridModel hybrid_model(ge_root_model); HybridModel hybrid_model(ge_root_model);
HybridModelBuilder hybrid_model_builder(hybrid_model); HybridModelBuilder hybrid_model_builder(hybrid_model);

ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), SUCCESS); ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), SUCCESS);
} }




Loading…
Cancel
Save