def METHOD_NAME():
    (
        client_num,
        train_data_num,
        test_data_num,
        train_data_global,
        test_data_global,
        data_local_num_dict,
        train_data_local_dict,
        test_data_local_dict,
        output_dim,
    ) = load_partition_data_federated_synthetic_1_1()
    f = open(train_file_path, "r")
    train_data = json.load(f)
    assert train_data["num_samples"] == list(data_local_num_dict.values())