EX2 Predict unittests