def test_remove_dataset(): fed_client = FederatedClient() dataset = "my_dataset" key = "string_dataset" fed_client.add_dataset(dataset, key) assert key in fed_client.datasets fed_client.remove_dataset(key) assert key not in fed_client.datasets
def test_remove_dataset(): # Create a client to execute federated learning fed_client = FederatedClient() # Create a dataset dataset = "my_dataset" key = "string_dataset" # Add new dataset fed_client.add_dataset(dataset, key) assert key in fed_client.datasets # Remove new dataset fed_client.remove_dataset(key) assert key not in fed_client.datasets