Test the reward function~