Test case for prediction sets added.