diff --git a/tests/st/testcase/test_tensorflow_parser.cc b/tests/st/testcase/test_tensorflow_parser.cc index 1b1389c..fa0db92 100644 --- a/tests/st/testcase/test_tensorflow_parser.cc +++ b/tests/st/testcase/test_tensorflow_parser.cc @@ -1124,7 +1124,7 @@ TEST_F(STestTensorflowParser, tensorflow_parserfrommemory_failed) ret = ge::aclgrphParseTensorFlow(modelFile.c_str(), parser_params, graph); ge::ComputeGraphPtr compute_graph = ge::GraphUtils::GetComputeGraph(graph); ret = modelParser.ParseFromMemory(data, size, compute_graph); - EXPECT_EQ(ret, INTERNAL_ERROR); + EXPECT_NE(ret, SUCCESS); } TEST_F(STestTensorflowParser, modelparser_parsefrommemory_success) diff --git a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc index 8910a13..671fefc 100644 --- a/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc +++ b/tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc @@ -1288,7 +1288,7 @@ TEST_F(UtestTensorflowParser, tensorflow_parserfrommemory_failed) ret = ge::aclgrphParseTensorFlow(modelFile.c_str(), parser_params, graph); ge::ComputeGraphPtr compute_graph = ge::GraphUtils::GetComputeGraph(graph); ret = modelParser.ParseFromMemory(data, size, compute_graph); - EXPECT_EQ(ret, INTERNAL_ERROR); + EXPECT_NE(ret, SUCCESS); } TEST_F(UtestTensorflowParser, modelparser_parsefrommemory_success)