def METHOD_NAME(self) -> None:
    """Test if the dataset works when multiple partitioners are specified."""
    num_train_partitions = 100
    num_test_partitions = 100
    dataset_fds = FederatedDataset(
        dataset=self.dataset_name,
        partitioners={
            "train": num_train_partitions,
            self.test_split: num_test_partitions,
        },
    )
    dataset_test_partition0 = dataset_fds.load_partition(0, self.test_split)
    dataset = datasets.load_dataset(self.dataset_name)
    self.assertEqual(
        len(dataset_test_partition0),
        len(dataset[self.test_split]) // num_test_partitions,
    )