from monai.transforms import ( PadListDataCollate, RandRotate, RandRotate90, RandRotate90d, RandRotated, RandSpatialCrop, RandSpatialCropd, RandZoom, RandZoomd, ) from monai.utils import set_determinism TESTS: List[Tuple] = [] for pad_collate in [pad_list_data_collate, PadListDataCollate()]: TESTS.append((dict, pad_collate, RandSpatialCropd("image", roi_size=[8, 7], random_size=True))) TESTS.append((dict, pad_collate, RandRotated("image", prob=1, range_x=np.pi, keep_size=False))) TESTS.append((dict, pad_collate, RandZoomd("image", prob=1, min_zoom=1.1, max_zoom=2.0, keep_size=False))) TESTS.append((dict, pad_collate, RandRotate90d("image", prob=1, max_k=2))) TESTS.append(
RandRotated, RandSpatialCrop, RandSpatialCropd, RandZoom, RandZoomd, ToTensor, ToTensord, ) from monai.utils import set_determinism TESTS: List[Tuple] = [] for pad_collate in [ lambda x: pad_list_data_collate(batch=x, method="end", mode="constant" ), PadListDataCollate(method="end", mode="constant"), ]: TESTS.append((dict, pad_collate, RandSpatialCropd("image", roi_size=[8, 7], random_size=True))) TESTS.append((dict, pad_collate, RandRotated("image", prob=1, range_x=np.pi, keep_size=False))) TESTS.append((dict, pad_collate, RandZoomd("image", prob=1, min_zoom=1.1, max_zoom=2.0, keep_size=False))) TESTS.append( (dict, pad_collate,