Added a unit test for the ANN