Added some JUnit tests for the classifier

only implemented testTrain