TST: Clean up tests in test_take.py
Utilizes `pytest` fixtures to cleanup the code significantly.

Initially thought this could be a good place for #21500, but that didn't turn out to be the case. 🤷‍♂️ 