From 37e4568ce3db3f19fff8febb883ee39a9299dd83 Mon Sep 17 00:00:00 2001 From: wjm Date: Mon, 12 Apr 2021 14:07:41 +0800 Subject: [PATCH] ut --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index c7fb2546..b5aac527 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -157,6 +157,7 @@ TEST_F(UtestGeHybrid, index_taskdefs_failed) { ge_root_model->SetModelName("test_name"); HybridModel hybrid_model(ge_root_model); HybridModelBuilder hybrid_model_builder(hybrid_model); + ASSERT_EQ(hybrid_model_builder.Build(), 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(graph); HybridModel hybrid_model(ge_root_model); HybridModelBuilder hybrid_model_builder(hybrid_model); + ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), SUCCESS); }