I often hear people say
I am taking GRE or I took the online test of Kaplan.
What I would like to know is that, is 'take' right here if i use it in the context of practice exercise in a Math book
I took the exercise of probability in X book, it was really challenging.
What would be the idiomatic way of saying it?