Browse Source

Add ut.

tags/v1.2.0
unknown 3 years ago
parent
commit
4766a04c4a
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      tests/ut/ge/single_op/stream_resource_unittest.cc

+ 2
- 0
tests/ut/ge/single_op/stream_resource_unittest.cc View File

@@ -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);
}


Loading…
Cancel
Save