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