@@ -83,7 +83,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_Transpose_and_Convolution) {
83
83
m.register_pass <ov::pass::InitNodeInfo>();
84
84
m.register_pass <ov::pass::FindBatch>();
85
85
m.run_passes (f);
86
- ASSERT_NO_THROW (check_rt_info (f));
86
+ OV_ASSERT_NO_THROW (check_rt_info (f));
87
87
88
88
const auto & shape = data->get_partial_shape ();
89
89
ASSERT_TRUE (!shape[0 ].get_symbol ()) << shape;
@@ -118,7 +118,7 @@ TEST(TransformationTests, AutoBatch_LabelPropagation_Convolution_Reshape) {
118
118
m.register_pass <ov::pass::InitNodeInfo>();
119
119
m.register_pass <ov::pass::FindBatch>();
120
120
m.run_passes (model);
121
- ASSERT_NO_THROW (check_rt_info (model));
121
+ OV_ASSERT_NO_THROW (check_rt_info (model));
122
122
123
123
const auto & shape = data->get_partial_shape ();
124
124
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -144,7 +144,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_SingleMultiply) {
144
144
m.register_pass <ov::pass::InitNodeInfo>();
145
145
m.register_pass <ov::pass::FindBatch>();
146
146
m.run_passes (f);
147
- ASSERT_NO_THROW (check_rt_info (f));
147
+ OV_ASSERT_NO_THROW (check_rt_info (f));
148
148
149
149
const auto & shape = data->get_partial_shape ();
150
150
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -174,7 +174,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_Two_Outputs) {
174
174
m.register_pass <ov::pass::InitNodeInfo>();
175
175
m.register_pass <ov::pass::FindBatch>();
176
176
m.run_passes (f);
177
- ASSERT_NO_THROW (check_rt_info (f));
177
+ OV_ASSERT_NO_THROW (check_rt_info (f));
178
178
179
179
const auto & shape = data->get_partial_shape ();
180
180
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -204,7 +204,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_TwoOutputsReversed) {
204
204
m.register_pass <ov::pass::InitNodeInfo>();
205
205
m.register_pass <ov::pass::FindBatch>();
206
206
m.run_passes (f);
207
- ASSERT_NO_THROW (check_rt_info (f));
207
+ OV_ASSERT_NO_THROW (check_rt_info (f));
208
208
209
209
const auto & shape = data->get_partial_shape ();
210
210
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -238,7 +238,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_IndependentBranchesConcated) {
238
238
m.register_pass <ov::pass::InitNodeInfo>();
239
239
m.register_pass <ov::pass::FindBatch>();
240
240
m.run_passes (f);
241
- ASSERT_NO_THROW (check_rt_info (f));
241
+ OV_ASSERT_NO_THROW (check_rt_info (f));
242
242
243
243
const auto & shape = data->get_partial_shape ();
244
244
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -271,7 +271,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_TwoConvNetwork) {
271
271
m.register_pass <ov::pass::InitNodeInfo>();
272
272
m.register_pass <ov::pass::FindBatch>();
273
273
m.run_passes (f);
274
- ASSERT_NO_THROW (check_rt_info (f));
274
+ OV_ASSERT_NO_THROW (check_rt_info (f));
275
275
276
276
const auto & shape = data->get_partial_shape ();
277
277
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -299,7 +299,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_NegativeTracking) {
299
299
m.register_pass <ov::pass::InitNodeInfo>();
300
300
m.register_pass <ov::pass::FindBatch>(false , false );
301
301
m.run_passes (f);
302
- ASSERT_NO_THROW (check_rt_info (f));
302
+ OV_ASSERT_NO_THROW (check_rt_info (f));
303
303
304
304
const auto & shape = data->get_partial_shape ();
305
305
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
@@ -319,7 +319,7 @@ TEST(TransformationTests, AutoBatch_FindBatch_AutoBatch_LabelPropagation_DO_deta
319
319
m.register_pass <ov::pass::InitNodeInfo>();
320
320
m.register_pass <ov::pass::FindBatch>(true );
321
321
m.run_passes (f);
322
- ASSERT_NO_THROW (check_rt_info (f));
322
+ OV_ASSERT_NO_THROW (check_rt_info (f));
323
323
324
324
const auto & shape = data->get_partial_shape ();
325
325
ASSERT_TRUE (shape[0 ].get_symbol ()) << shape;
0 commit comments