()
| 119 | |
| 120 | |
| 121 | def test_data_loader(): |
| 122 | ( |
| 123 | client_num, |
| 124 | train_data_num, |
| 125 | test_data_num, |
| 126 | train_data_global, |
| 127 | test_data_global, |
| 128 | data_local_num_dict, |
| 129 | train_data_local_dict, |
| 130 | test_data_local_dict, |
| 131 | output_dim, |
| 132 | ) = load_partition_data_federated_synthetic_1_1() |
| 133 | f = open(train_file_path, "r") |
| 134 | train_data = json.load(f) |
| 135 | assert train_data["num_samples"] == list(data_local_num_dict.values()) |
| 136 | |
| 137 | |
| 138 | if __name__ == "__main__": |
no test coverage detected