def METHOD_NAME(self):
    for p in TEST_NDARRAYS:
        transforms = RandFlipd(["image", "label"], prob=1.0)
        tta = TestTimeAugmentation(transforms, batch_size=5, num_workers=0, inferrer_fn=lambda x: x)
        tta(self.get_data(1, (20, 20), data_type=p))