Yea, you can not really predict the future, if you can not imagine. And without it being able to predict the future, a dog would never be able to be trained.

So, any animal that you can teach something through potential treat must be able to imagine, I would assume.