(kinds, types, op, shape)
| 964 | |
| 965 | |
| 966 | def check_raises(kinds, types, op, shape): |
| 967 | if isinstance(op, tuple): |
| 968 | dali_op = op[0] |
| 969 | else: |
| 970 | dali_op = op |
| 971 | iterator = iter(ExternalInputIterator(batch_size, shape, types, kinds)) |
| 972 | pipe = ExprOpPipeline( |
| 973 | kinds, types, iterator, dali_op, batch_size=batch_size, num_threads=2, device_id=0 |
| 974 | ) |
| 975 | pipe.run() |
| 976 | |
| 977 | |
| 978 | def check_raises_re(kinds, types, op, shape, _, msg): |
no test coverage detected