def METHOD_NAME(self):
    store = FakeStore()
    dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
    out_tensor = torch.ones(3, 3)
    in_tensor = torch.ones(3, 3)
    output_split = [1, 1]
    input_split = [1, 1]
    dist.all_to_all_single(out_tensor, in_tensor, output_split, input_split)
    self.assertEqual(tuple(out_tensor.shape), (3, 3))