Wrote a simple unit test for predictor base