(
file_root,
batch_size,
ndim,
dtype,
device,
fortran_order=False,
file_filter="*.npy",
roi_start=None,
rel_roi_start=None,
roi_end=None,
rel_roi_end=None,
roi_shape=None,
rel_roi_shape=None,
roi_axes=None,
out_of_bounds_policy=None,
fill_value=None,
)
| 561 | |
| 562 | |
| 563 | def _testimpl_numpy_reader_roi( |
| 564 | file_root, |
| 565 | batch_size, |
| 566 | ndim, |
| 567 | dtype, |
| 568 | device, |
| 569 | fortran_order=False, |
| 570 | file_filter="*.npy", |
| 571 | roi_start=None, |
| 572 | rel_roi_start=None, |
| 573 | roi_end=None, |
| 574 | rel_roi_end=None, |
| 575 | roi_shape=None, |
| 576 | rel_roi_shape=None, |
| 577 | roi_axes=None, |
| 578 | out_of_bounds_policy=None, |
| 579 | fill_value=None, |
| 580 | ): |
| 581 | default_axes = list(range(ndim)) |
| 582 | pipe = numpy_reader_roi_pipe( |
| 583 | file_root=file_root, |
| 584 | file_filter=file_filter, |
| 585 | device=device, |
| 586 | roi_start=roi_start, |
| 587 | rel_roi_start=rel_roi_start, |
| 588 | roi_end=roi_end, |
| 589 | rel_roi_end=rel_roi_end, |
| 590 | roi_shape=roi_shape, |
| 591 | rel_roi_shape=rel_roi_shape, |
| 592 | roi_axes=roi_axes, |
| 593 | default_axes=default_axes, |
| 594 | out_of_bounds_policy=out_of_bounds_policy, |
| 595 | fill_value=fill_value, |
| 596 | batch_size=batch_size, |
| 597 | ) |
| 598 | |
| 599 | try: |
| 600 | roi_out, sliced_out = pipe.run() |
| 601 | for i in range(batch_size): |
| 602 | roi_arr = to_array(roi_out[i]) |
| 603 | sliced_arr = to_array(sliced_out[i]) |
| 604 | assert_array_equal(roi_arr, sliced_arr) |
| 605 | finally: |
| 606 | del pipe |
| 607 | |
| 608 | |
| 609 | def _testimpl_numpy_reader_roi_empty_axes( |
no test coverage detected