| 412 | self.assertTupleEqual(result.shape, spatial_size[::-1]) |
| 413 | |
| 414 | def test_kwargs(self): |
| 415 | spatial_size = (32, 64, 128) |
| 416 | test_image = np.random.rand(*spatial_size) |
| 417 | with tempfile.TemporaryDirectory() as tempdir: |
| 418 | filename = os.path.join(tempdir, "test_image.nii.gz") |
| 419 | itk_np_view = itk.image_view_from_array(test_image) |
| 420 | itk.imwrite(itk_np_view, filename) |
| 421 | |
| 422 | loader = LoadImage(image_only=True) |
| 423 | reader = ITKReader(fallback_only=False) |
| 424 | loader.register(reader) |
| 425 | result = loader(filename) |
| 426 | |
| 427 | reader = ITKReader() |
| 428 | img = reader.read(filename, fallback_only=False) |
| 429 | result_raw = reader.get_data(img) |
| 430 | result_raw = MetaTensor.ensure_torch_and_prune_meta(*result_raw) |
| 431 | self.assertTupleEqual(result.shape, result_raw.shape) |
| 432 | |
| 433 | def test_my_reader(self): |
| 434 | """test customised readers""" |