I think the test was provided by some company. There was nothing magical about it though - part of it looked mostly like an IQ test, and there were math questions and short reading comprehension tasks. It was fairly easy and the biggest difficulty I thought was the short amount of time we had to complete it.

I suspect any reasonable test in that general direction would do the job. I don't think they use it to specifically hire the people who achieve the very best scores, but rather to not hire the ones who perform poorly.

They are usually called "Psychotechnical tests", and their use is standard in many industries. Though judging for some people I've seen passing them, I'd say that most of them are borderline useless, or not really used.