(self)
| 60 | class TestImageDataset(unittest.TestCase): |
| 61 | |
| 62 | def test_use_case(self): |
| 63 | with tempfile.TemporaryDirectory() as tempdir: |
| 64 | img_ = nib.Nifti1Image(np.random.randint(0, 2, size=(20, 20, 20)).astype(float), np.eye(4)) |
| 65 | seg_ = nib.Nifti1Image(np.random.randint(0, 2, size=(20, 20, 20)).astype(float), np.eye(4)) |
| 66 | img_name, seg_name = os.path.join(tempdir, "img.nii.gz"), os.path.join(tempdir, "seg.nii.gz") |
| 67 | nib.save(img_, img_name) |
| 68 | nib.save(seg_, seg_name) |
| 69 | img_list, seg_list = [img_name], [seg_name] |
| 70 | |
| 71 | img_xform = _TestCompose([EnsureChannelFirst(), Spacing(pixdim=(1.5, 1.5, 3.0)), RandAdjustContrast()]) |
| 72 | seg_xform = _TestCompose([EnsureChannelFirst(), Spacing(pixdim=(1.5, 1.5, 3.0), mode="nearest")]) |
| 73 | img_dataset = ImageDataset( |
| 74 | image_files=img_list, |
| 75 | seg_files=seg_list, |
| 76 | transform=img_xform, |
| 77 | seg_transform=seg_xform, |
| 78 | image_only=False, |
| 79 | transform_with_metadata=True, |
| 80 | ) |
| 81 | self.assertTupleEqual(img_dataset[0][0].shape, (1, 14, 14, 7)) |
| 82 | self.assertTupleEqual(img_dataset[0][1].shape, (1, 14, 14, 7)) |
| 83 | |
| 84 | def test_dataset(self): |
| 85 | with tempfile.TemporaryDirectory() as tempdir: |
nothing calls this directly
no test coverage detected