When you are studying for a test and doing a lot of exercises. Why not train by your imagination.
The idea goes as follows. Take an exercise and before you attempt it. Make one or two new exercises from your imagination and the target exercise as your guide. Here you can transform the exercise to a much easier variant or a little harder one.
After solving both first you will have trained your imagination. The target exercise is meant for verification.