Add unit tests for PredictionModel.