From 4766a04c4a02250acc1da3b03f7376ca77e9cf04 Mon Sep 17 00:00:00 2001 From: unknown Date: Mon, 1 Mar 2021 09:14:37 +0800 Subject: [PATCH] Add ut. --- tests/ut/ge/single_op/stream_resource_unittest.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/ut/ge/single_op/stream_resource_unittest.cc b/tests/ut/ge/single_op/stream_resource_unittest.cc index 9332be53..e07fc39d 100644 --- a/tests/ut/ge/single_op/stream_resource_unittest.cc +++ b/tests/ut/ge/single_op/stream_resource_unittest.cc @@ -66,6 +66,8 @@ TEST_F(UtestStreamResource, test_build_op) { res.op_map_[0].reset(single_op); res.dynamic_op_map_[1].reset(dynamic_single_op); + EXPECT_EQ(res.GetOperator(0), nullptr); + EXPECT_EQ(res.GetDynamicOperator(1), nullptr); EXPECT_EQ(res.BuildOperator(model_data, &single_op, 0), SUCCESS); EXPECT_EQ(res.BuildDynamicOperator(model_data, &dynamic_single_op, 1), SUCCESS); }