MNIST Using Federated Learning - Test Code - View it on GitHub