(self)
| 91 | """Test collation for of random transformations with prob == 0 and 1.""" |
| 92 | |
| 93 | def setUp(self): |
| 94 | if not has_nib: |
| 95 | self.skipTest("nibabel required for test_inverse") |
| 96 | |
| 97 | set_determinism(seed=0) |
| 98 | |
| 99 | b_size = 11 |
| 100 | im_fname, seg_fname = (make_nifti_image(i) for i in create_test_image_3d(101, 100, 107)) |
| 101 | load_ims = Compose([LoadImaged(KEYS), EnsureChannelFirstd(KEYS, channel_dim="no_channel")]) |
| 102 | self.data_3d = [load_ims({"image": im_fname, "label": seg_fname}) for _ in range(b_size)] |
| 103 | |
| 104 | b_size = 8 |
| 105 | im_fname, seg_fname = (make_nifti_image(i) for i in create_test_image_2d(62, 37, rad_max=10)) |
| 106 | load_ims = Compose([LoadImaged(KEYS), EnsureChannelFirstd(KEYS, channel_dim="no_channel")]) |
| 107 | self.data_2d = [load_ims({"image": im_fname, "label": seg_fname}) for _ in range(b_size)] |
| 108 | |
| 109 | self.batch_size = 7 |
| 110 | |
| 111 | def tearDown(self): |
| 112 | set_determinism(seed=None) |
nothing calls this directly
no test coverage detected