Added tests for Prediction class.