(src_dims, rel_shape, shapes, expected_out_shapes)
| 332 | |
| 333 | |
| 334 | def _testimpl_reshape_src_dims_arg(src_dims, rel_shape, shapes, expected_out_shapes): |
| 335 | batch_size = len(shapes) |
| 336 | pipe = reshape_pipe( |
| 337 | batch_size=batch_size, |
| 338 | num_threads=1, |
| 339 | device_id=0, |
| 340 | shapes=shapes, |
| 341 | src_dims=src_dims, |
| 342 | rel_shape=rel_shape, |
| 343 | ) |
| 344 | for _ in range(3): |
| 345 | outs = pipe.run() |
| 346 | for i in range(batch_size): |
| 347 | out_arr = np.array(outs[0][i]) |
| 348 | assert out_arr.shape == expected_out_shapes[i] |
| 349 | |
| 350 | |
| 351 | def test_reshape_src_dims_arg(): |
nothing calls this directly
no test coverage detected