Wrote a basic unit test for the neural network training.